質問編集履歴
2
コードのインデントがきちんと表示されていなかったため
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -38,7 +38,7 @@
|
|
|
38
38
|
img, y = self.horizontal_flip(img, y)
|
|
39
39
|
if self.vflip_prob > 0:
|
|
40
40
|
img, y = self.vertical_flip(img, y)
|
|
41
|
-
y = self.bbox_util.assign_boxes(y)
|
|
41
|
+
y = self.bbox_util.assign_boxes(y)
|
|
42
42
|
inputs.append(img)
|
|
43
43
|
targets.append(y)
|
|
44
44
|
if len(targets) == self.batch_size:
|
1
コードのインデントがきちんと表示されていなかったため
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -6,6 +6,7 @@
|
|
|
6
6
|
参考にしたサイト①で紹介されている方法で学習用のデータを作製したのち
|
|
7
7
|
参考にしたサイト②のSSD_train_custom.pyを実行したのですが、
|
|
8
8
|
(以下SSD_train_custom.pyのコード
|
|
9
|
+
```
|
|
9
10
|
...
|
|
10
11
|
def generate(self, train=True):
|
|
11
12
|
while True:
|
|
@@ -37,7 +38,7 @@
|
|
|
37
38
|
img, y = self.horizontal_flip(img, y)
|
|
38
39
|
if self.vflip_prob > 0:
|
|
39
40
|
img, y = self.vertical_flip(img, y)
|
|
40
|
-
y = self.bbox_util.assign_boxes(y)
|
|
41
|
+
y = self.bbox_util.assign_boxes(y)```
|
|
41
42
|
inputs.append(img)
|
|
42
43
|
targets.append(y)
|
|
43
44
|
if len(targets) == self.batch_size:
|
|
@@ -99,9 +100,10 @@
|
|
|
99
100
|
elapsed_time = time.time() - start
|
|
100
101
|
print ("elapsed_time:{0}".format(elapsed_time) + "[sec]")
|
|
101
102
|
....
|
|
103
|
+
```
|
|
102
104
|
)
|
|
103
105
|
|
|
104
|
-
|
|
106
|
+
以下のようなエラーが出てしまいました。
|
|
105
107
|
|
|
106
108
|
Exception in thread Thread-9:
|
|
107
109
|
Traceback (most recent call last):
|