Rename img_size to imgsz (#86)

2022-12-24 00:39:09 +01:00
parent ae2443c210
commit 6432afc5f9
25 changed files with 98 additions and 98 deletions
--- a/ultralytics/yolo/v8/classify/val.py
+++ b/ultralytics/yolo/v8/classify/val.py
@ -28,7 +28,7 @@ class ClassificationValidator(BaseValidator):
        return {"top1": top1, "top5": top5, "fitness": top5}

    def get_dataloader(self, dataset_path, batch_size):
-        return build_classification_dataloader(path=dataset_path, imgsz=self.args.img_size, batch_size=batch_size)
+        return build_classification_dataloader(path=dataset_path, imgsz=self.args.imgsz, batch_size=batch_size)

    @property
    def metric_keys(self):