Update yolo_training.py

3229018240 · Jan 17, 2021 · 5e03500 · 5e03500
1 parent a569426
commit 5e03500
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nets/yolo_training.py b/nets/yolo_training.py
@@ -515,7 +515,7 @@ def generate(self, train=True):
                 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)