From 8fea4514fb828ab9738d5583c68740b9d7e5ff2d Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 8 Oct 2019 14:35:43 +0200
Subject: [PATCH] updates

---
 utils/datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/datasets.py b/utils/datasets.py
index 449f801f..2454dea9 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -405,7 +405,7 @@ class LoadImagesAndLabels(Dataset):  # for training/testing
         img_path = self.img_files[index]
         label_path = self.label_files[index]
 
-        mosaic = True  # load 4 images at a time into a mosaic
+        mosaic = True and not self.augment  # load 4 images at a time into a mosaic (only during training)
         if mosaic:
             # Load mosaic
             img, labels = load_mosaic(self, index)