From 86f266f32656897084aefeff02b514ad6244d437 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Mon, 18 Jan 2021 00:17:55 +0800
Subject: [PATCH] Update train_with_tensorboard.py

---
 train_with_tensorboard.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train_with_tensorboard.py b/train_with_tensorboard.py
index 8d14e2c..93a3c2a 100644
--- a/train_with_tensorboard.py
+++ b/train_with_tensorboard.py
@@ -272,7 +272,7 @@ if __name__ == "__main__":
             gen = Generator(Batch_size, lines[:num_train],
                             (input_shape[0], input_shape[1])).generate(train=True, mosaic = mosaic)
             gen_val = Generator(Batch_size, lines[num_train:],
-                            (input_shape[0], input_shape[1])).generate(train=False, mosaic = mosaic)
+                            (input_shape[0], input_shape[1])).generate(train=False, mosaic = False)
 
         epoch_size = max(1, num_train//Batch_size)
         epoch_size_val = num_val//Batch_size
@@ -309,7 +309,7 @@ if __name__ == "__main__":
             gen = Generator(Batch_size, lines[:num_train],
                             (input_shape[0], input_shape[1])).generate(train=True, mosaic = mosaic)
             gen_val = Generator(Batch_size, lines[num_train:],
-                            (input_shape[0], input_shape[1])).generate(train=False, mosaic = mosaic)
+                            (input_shape[0], input_shape[1])).generate(train=False, mosaic = False)
 
         epoch_size = max(1, num_train//Batch_size)
         epoch_size_val = num_val//Batch_size
-- 
GitLab