This commit is contained in:
Glenn Jocher 2019-10-08 14:35:43 +02:00
parent 69745d8b8e
commit 8fea4514fb
1 changed files with 1 additions and 1 deletions

View File

@ -405,7 +405,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
img_path = self.img_files[index] img_path = self.img_files[index]
label_path = self.label_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: if mosaic:
# Load mosaic # Load mosaic
img, labels = load_mosaic(self, index) img, labels = load_mosaic(self, index)