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

6
ultralytics/yolo/v8/segment/predict.py
View file

@ -102,11 +102,11 @@ class SegmentationPredictor(DetectionPredictor):
@hydra.main(version_base=None, config_path=DEFAULT_CONFIG.parent, config_name=DEFAULT_CONFIG.name)
def predict(cfg):
cfg.model = cfg.model or "n.pt"
sz = cfg.img_size
sz = cfg.imgsz
if type(sz) != int: # recieved listConfig
cfg.img_size = [sz[0], sz[0]] if len(cfg.img_size) == 1 else [sz[0], sz[1]] # expand
cfg.imgsz = [sz[0], sz[0]] if len(cfg.imgsz) == 1 else [sz[0], sz[1]] # expand
else:
cfg.img_size = [sz, sz]
cfg.imgsz = [sz, sz]
predictor = SegmentationPredictor(cfg)
predictor()

2
ultralytics/yolo/v8/segment/train.py
View file

@ -243,7 +243,7 @@ def train(cfg):
if __name__ == "__main__":
"""
CLI usage:
python ultralytics/yolo/v8/segment/train.py model=yolov5n-seg.yaml data=coco128-segments epochs=100 img_size=640
python ultralytics/yolo/v8/segment/train.py model=yolov5n-seg.yaml data=coco128-segments epochs=100 imgsz=640
TODO:
Direct cli support, i.e, yolov8 classify_train args.epochs 10