我正在尝试在yolov5中训练一个自定义数据集。因此,我试图运行它的图像大小为640x480,但它不工作。
python3 /yolov5 5/yolov5 5/tra.py-img-size 640 480 -批8-数据data.yaml - yolov5s.pt -缓存
usage: train.py [-h] [--weights WEIGHTS] [--cfg CFG] [--data DATA] [--hyp HYP] [--epochs EPOCHS]
[--batch-size BATCH_SIZE] [--imgsz IMGSZ] [--rect] [--resume [RESUME]] [--nosave]
[--noval] [--noautoanchor] [--noplots] [--evolve [EVOLVE]] [--bucket BUCKET]
[--cache [CACHE]] [--image-weights] [--device DEVICE] [--multi-scale]
[--single-cls] [--optimizer {SGD,Adam,AdamW}] [--sync-bn] [--workers WORKERS]
[--project PROJECT] [--name NAME] [--exist-ok] [--quad] [--cos-lr]
[--label-smoothing LABEL_SMOOTHING] [--patience PATIENCE]
[--freeze FREEZE [FREEZE ...]] [--save-period SAVE_PERIOD]
[--local_rank LOCAL_RANK] [--entity ENTITY] [--upload_dataset [UPLOAD_DATASET]]
[--bbox_interval BBOX_INTERVAL] [--artifact_alias ARTIFACT_ALIAS]
train.py: error: unrecognized arguments: 480
def parse_opt(known=False):parser.add_argument(‘-imgsz’,‘-img’,‘-img- size’,type=int,default=640,help='train,val图像大小(像素)‘) (https://github.com/ultralytics/yolov5.git)
发布于 2022-11-26 08:09:37
--img-size
只需要一个论点。使用:
python3 /YOLOv5/yolov5/train.py --img-size 640 --batch 8 --epochs 300 --data data.yaml --weights yolov5s.pt --cache
图像的高度将相应调整,尊重高宽比和步幅需要。
https://stackoverflow.com/questions/74457702
复制相似问题