From d92755323135ff690db91b679e707d53d0b4bd14 Mon Sep 17 00:00:00 2001
From: JiaQi Xu <47347516+bubbliiiing@users.noreply.github.com>
Date: Tue, 12 May 2020 11:02:23 +0800
Subject: [PATCH] Update train.py

---
 train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train.py b/train.py
index 7dcb889..4aa3310 100644
--- a/train.py
+++ b/train.py
@@ -44,14 +44,14 @@ def data_generator(annotation_lines, batch_size, input_shape, anchors, num_class
                 np.random.shuffle(annotation_lines)
             if mosaic:
                 if flag and (i+4) < n:
-                    image, box = get_random_data_with_Mosaic(annotation_lines[i:i+4], input_shape, random=True)
+                    image, box = get_random_data_with_Mosaic(annotation_lines[i:i+4], input_shape)
                     i = (i+4) % n
                 else:
-                    image, box = get_random_data(annotation_lines[i], input_shape, random=True)
+                    image, box = get_random_data(annotation_lines[i], input_shape)
                     i = (i+1) % n
                 flag = bool(1-flag)
             else:
-                image, box = get_random_data(annotation_lines[i], input_shape, random=True)
+                image, box = get_random_data(annotation_lines[i], input_shape)
                 i = (i+1) % n
             image_data.append(image)
             box_data.append(box)
-- 
GitLab