From 5e03500f5171cf15b9cf3892085535924b792b64 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Mon, 18 Jan 2021 01:05:53 +0800
Subject: [PATCH] Update yolo_training.py

---
 nets/yolo_training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nets/yolo_training.py b/nets/yolo_training.py
index 6bc9610..f9fac9c 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)
-- 
GitLab