trying to remove mosaic

nikitabelooussovbtis · Nov 1, 2022 · 7f68636 · 7f68636
1 parent 258acda
commit 7f68636
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/train.py b/train.py
@@ -224,7 +224,7 @@ def train(hyp, opt, device, callbacks):  # hyp is path/to/hyp.yaml or hyp dictio
                                               gs,
                                               single_cls,
                                               hyp=hyp,
-                                              augment=False,
+                                              augment=True,
                                               cache=None if opt.cache == 'val' else opt.cache,
                                               rect=opt.rect,
                                               rank=LOCAL_RANK,
@@ -309,7 +309,7 @@ def train(hyp, opt, device, callbacks):  # hyp is path/to/hyp.yaml or hyp dictio
                                                   gs,
                                                   single_cls,
                                                   hyp=hyp,
-                                                  augment=False,
+                                                  augment=True,
                                                   cache=None if opt.cache == 'val' else opt.cache,
                                                   rect=opt.rect,
                                                   rank=LOCAL_RANK,

diff --git a/utils/dataloaders.py b/utils/dataloaders.py
@@ -403,7 +403,7 @@ def __init__(self,
                  path,
                  img_size=640,
                  batch_size=16,
-                 augment=False,
+                 augment=True,
                  hyp=None,
                  rect=False,
                  image_weights=False,
@@ -418,6 +418,7 @@ def __init__(self,
         self.image_weights = image_weights
         self.rect = False if image_weights else rect
         self.mosaic = self.augment and not self.rect  # load 4 images at a time into a mosaic (only during training)
+        self.mosaic = False
         self.mosaic_border = [-img_size // 2, -img_size // 2]
         self.stride = stride
         self.path = path