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)