remains mixup and copy_paste

8 months ago · ded2f84b97
parent ed6b483361
commit ded2f84b97
1 changed files with 2 additions and 2 deletions
--- a/ultralytics/data/dataset.py
+++ b/ultralytics/data/dataset.py
@ -175,8 +175,8 @@ class YOLODataset(BaseDataset):
    def close_mosaic(self, hyp):
        """Sets mosaic, copy_paste and mixup options to 0.0 and builds transformations."""
        hyp.mosaic = 0.0  # set mosaic ratio=0.0
-        hyp.copy_paste = 0.0  # keep the same behavior as previous v8 close-mosaic
-        hyp.mixup = 0.0  # keep the same behavior as previous v8 close-mosaic
+        # hyp.copy_paste = 0.0  # keep the same behavior as previous v8 close-mosaic
+        # hyp.mixup = 0.0  # keep the same behavior as previous v8 close-mosaic
        self.transforms = self.build_transforms(hyp)

    def update_labels_info(self, label):