removed aug

nikitabelooussovbtis · Nov 1, 2022 · de179d0 · de179d0
1 parent 7179aa5
commit de179d0
Show file tree

Hide file tree

Showing 2 changed files with 2 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=True,
+                                              augment=False,
                                               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=True,
+                                                  augment=False,
                                                   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
@@ -423,7 +423,6 @@ def __init__(self,
         self.stride = stride
         self.path = path
         self.albumentations = Albumentations() if augment else None
-        self.albumentations = False
         try:
             f = []  # image files
             for p in path if isinstance(path, list) else [path]: