HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Rename img_size to imgsz (#86)

Browse source
This commit is contained in:
Glenn Jocher 2022-12-24 00:39:09 +01:00 • committed by GitHub
parent ae2443c210
commit 6432afc5f9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
25 changed files with 98 additions and 98 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/v8/classify/val.py
View file

@ -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):