|
|
|
@ -145,6 +145,7 @@ class YOLODataset(BaseDataset):
|
|
|
|
# TODO: use hyp config to set all these augmentations
|
|
|
|
# TODO: use hyp config to set all these augmentations
|
|
|
|
def build_transforms(self, hyp=None):
|
|
|
|
def build_transforms(self, hyp=None):
|
|
|
|
"""Builds and appends transforms to the list."""
|
|
|
|
"""Builds and appends transforms to the list."""
|
|
|
|
|
|
|
|
self.augment = False
|
|
|
|
if self.augment:
|
|
|
|
if self.augment:
|
|
|
|
hyp.mosaic = hyp.mosaic if self.augment and not self.rect else 0.0
|
|
|
|
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
|
|
|
|
hyp.mixup = hyp.mixup if self.augment and not self.rect else 0.0
|
|
|
|
|
