From c5d7b643277927269b68a2488ef976d36acfaf6c Mon Sep 17 00:00:00 2001
From: Francesco Mattioli <Francesco.mttl@gmail.com>
Date: Tue, 3 Sep 2024 13:24:08 +0200
Subject: [PATCH] fixed augment

---
 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 df3425037c..7eb3d3817b 100644
--- 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,