updates
This commit is contained in:
parent
69745d8b8e
commit
8fea4514fb
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue