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]
|
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)
|
||||||
|
|
Loading…
Reference in New Issue