質問編集履歴
2
コードの編集
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -5,73 +5,12 @@
|
|
|
5
5
|
仕様変更などの問題かもしれませんが教えていただけますでしょうか。
|
|
6
6
|
|
|
7
7
|
何か足りないものがあったら編集しますのでコメントお願いします。
|
|
8
|
+
保存する部分は以下のようになっています。
|
|
8
9
|
```python
|
|
9
|
-
# coding: UTF-8
|
|
10
|
-
|
|
11
|
-
import cv2
|
|
12
|
-
import pandas as pd
|
|
13
|
-
import glob, os, sys
|
|
14
|
-
from datetime import datetime
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
NUM_CLASSES = 3
|
|
18
|
-
|
|
19
|
-
MOVIE_EXTENSION = "mp4"
|
|
20
|
-
IMAGE_EXTENSION = "jpg"
|
|
21
|
-
# リサイズ後の画像サイズ
|
|
22
|
-
IMAGE_SIZE = int(sys.argv[1])
|
|
23
|
-
|
|
24
|
-
|
|
25
|
-
# 移動先の場所
|
|
26
|
-
DESTINATION_PATH = "D:\Users\Sohto\Documents\R01\RE\03.ソースコード\明度測定用_" + sys.argv[1]
|
|
27
|
-
# 動画が入っているフォルダがある場所
|
|
28
|
-
SOURCE_PATH = "M:\R.01_sapoin\Chiku\01.内視鏡\2019_11_08"
|
|
29
|
-
|
|
30
|
-
CLASS_FILE = "class.csv"
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
print("動画拡張子:" + MOVIE_EXTENSION + " 画像拡張子:" + IMAGE_EXTENSION)
|
|
34
|
-
print("移動先:" + DESTINATION_PATH)
|
|
35
|
-
|
|
36
|
-
if not os.path.exists(DESTINATION_PATH):
|
|
37
|
-
os.mkdir(DESTINATION_PATH)
|
|
38
|
-
|
|
39
|
-
for class_label in range(NUM_CLASSES):
|
|
40
|
-
class_dir = DESTINATION_PATH + "/" + str(class_label)
|
|
41
|
-
if not os.path.exists(class_dir):
|
|
42
|
-
os.mkdir(class_dir)
|
|
43
|
-
|
|
44
|
-
df = pd.read_csv(CLASS_FILE)
|
|
45
|
-
class_dict = {x[0]: x[1] for x in df.itertuples(index=False)}
|
|
46
|
-
|
|
47
|
-
files = os.listdir(SOURCE_PATH)
|
|
48
|
-
source_file = [f for f in files if os.path.isfile(os.path.join(SOURCE_PATH, f))]
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
10
|
+
while(cap.isOpened()):
|
|
52
|
-
split = file.split("_")
|
|
53
|
-
number = int(split[0])
|
|
54
|
-
|
|
55
|
-
class_dir = DESTINATION_PATH + "/" + str(class_dict[number])
|
|
56
|
-
|
|
57
|
-
|
|
11
|
+
# ret:動画が続いているかTrue or False, frame:そのフレームの画像
|
|
58
|
-
if not os.path.exists(image_dir):
|
|
59
|
-
os.mkdir(image_dir)
|
|
60
|
-
|
|
61
|
-
# scope version
|
|
62
|
-
movie_file = file
|
|
63
|
-
|
|
64
|
-
if movie_file == []:
|
|
65
|
-
continue
|
|
66
|
-
|
|
67
|
-
print(movie_file)
|
|
68
|
-
|
|
69
|
-
cap = cv2.VideoCapture(SOURCE_PATH + "/" + movie_file)
|
|
70
|
-
count = 0
|
|
71
|
-
|
|
72
|
-
while True:
|
|
73
12
|
ret, frame = cap.read()
|
|
74
|
-
if ret:
|
|
13
|
+
if ret == True:
|
|
75
14
|
count += 1
|
|
76
15
|
|
|
77
16
|
h, w, c = frame.shape
|
|
@@ -81,6 +20,7 @@
|
|
|
81
20
|
frame = cv2.resize(frame, (IMAGE_SIZE, IMAGE_SIZE), cv2.INTER_AREA)
|
|
82
21
|
# 画像を保存
|
|
83
22
|
cv2.imwrite(image_dir + "/" + str("{0:08d}".format(count)) + '.' + IMAGE_EXTENSION, frame)
|
|
23
|
+
print('Save', image_dir + "/" + str("{0:08d}".format(count)) + '.' + IMAGE_EXTENSION)
|
|
84
24
|
else:
|
|
85
25
|
count = 0
|
|
86
26
|
break
|
1
文言の追加
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -3,6 +3,8 @@
|
|
|
3
3
|
以前は画像生成ができていたのですが、おとといあたりから急に
|
|
4
4
|
生成されなくなりました。
|
|
5
5
|
仕様変更などの問題かもしれませんが教えていただけますでしょうか。
|
|
6
|
+
|
|
7
|
+
何か足りないものがあったら編集しますのでコメントお願いします。
|
|
6
8
|
```python
|
|
7
9
|
# coding: UTF-8
|
|
8
10
|
|