Update yolo_training.py

5e03500f · Bubbliiiing · GitHub · a5694268 · 5e03500f
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

nets/yolo_training.py nets/yolo_training.py +1 -1

未找到文件。
--- a/nets/yolo_training.py
+++ b/nets/yolo_training.py
@@ -515,7 +515,7 @@ class Generator(object):
                if train:
                    img,y=self.get_random_data(annotation_line, self.image_size[0:2])
                else:
-                    img,y=self.get_random_data(annotation_line, self.image_size[0:2], False)
+                    img,y=self.get_random_data(annotation_line, self.image_size[0:2], random=False)

                if len(y)!=0:
                    boxes = np.array(y[:,:4],dtype=np.float32)