fixed augment

6 months ago · c5d7b64327
parent 5b3e60b452
commit c5d7b64327
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/data/build.py
+++ b/ultralytics/data/build.py
@ -88,7 +88,7 @@ def build_yolo_dataset(cfg, img_path, batch, data, mode="train", rect=False, str
        img_path=img_path,
        imgsz=cfg.imgsz,
        batch_size=batch,
-        augment=mode == "train",  # augmentation
+        augment=cfg.augment if mode=="train" else False,  # augmentation
        hyp=cfg,  # TODO: probably add a get_hyps_from_cfg function
        rect=cfg.rect or rect,  # rectangular batches
        cache=cfg.cache or None,