disabled augmentations

single_channel
Samuel Dobron 8 months ago
parent 82b97d5d4c
commit d15113ed0a
Signed by: xdobro23
GPG Key ID: 5C0F45DB63204F86

@ -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

Loading…
Cancel
Save