Rename img_size to imgsz (#86)

2022-12-24 00:39:09 +01:00 · 2022-12-24 00:39:09 +01:00 · 6432afc5f9
commit 6432afc5f9
parent ae2443c210
25 changed files with 98 additions and 98 deletions
--- a/ultralytics/yolo/data/build.py
+++ b/ultralytics/yolo/data/build.py
@ -62,7 +62,7 @@ def build_dataloader(cfg, batch_size, img_path, stride=32, label_path=None, rank
        dataset = YOLODataset(
            img_path=img_path,
            label_path=label_path,
-            img_size=cfg.img_size,
+            imgsz=cfg.imgsz,
            batch_size=batch_size,
            augment=True if mode == "train" else False,  # augmentation
            hyp=cfg,  # TODO: probably add a get_hyps_from_cfg function