no augmentations

2 weeks ago · 3209a2ccf2
parent f9a4aadfc6
commit 3209a2ccf2
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/data/build.py
+++ b/ultralytics/data/build.py
@ -93,7 +93,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=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,