diff --git a/nets/yolo_training.py b/nets/yolo_training.py index 6bc9610d6854726b03be9c655f83b8bf53c45061..f9fac9c73a1134a387f72becda673b379ad40455 100644 --- 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)