From 3209a2ccf270a760f783d01ccee8e1e1b5701cc9 Mon Sep 17 00:00:00 2001 From: Laughing-q <1185102784@qq.com> Date: Thu, 14 Nov 2024 17:23:01 +0800 Subject: [PATCH] no augmentations --- ultralytics/data/build.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/data/build.py b/ultralytics/data/build.py index 30f5aaa89d..7f98af9ba7 100644 --- 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,