From d15113ed0aeb1bc5fa6002611882eeb8edd5a0a3 Mon Sep 17 00:00:00 2001 From: Samuel Dobron Date: Mon, 29 Apr 2024 16:11:57 +0200 Subject: [PATCH] disabled augmentations --- ultralytics/data/dataset.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ultralytics/data/dataset.py b/ultralytics/data/dataset.py index 6d62039..041d8eb 100644 --- a/ultralytics/data/dataset.py +++ b/ultralytics/data/dataset.py @@ -145,6 +145,7 @@ class YOLODataset(BaseDataset): # TODO: use hyp config to set all these augmentations def build_transforms(self, hyp=None): """Builds and appends transforms to the list.""" + self.augment = False if self.augment: hyp.mosaic = hyp.mosaic if self.augment and not self.rect else 0.0 hyp.mixup = hyp.mixup if self.augment and not self.rect else 0.0