首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >yolov5训练中的图像大小

yolov5训练中的图像大小
EN

Stack Overflow用户
提问于 2022-11-16 09:00:40
回答 1查看 71关注 0票数 0

我正在尝试在yolov5中训练一个自定义数据集。因此,我试图运行它的图像大小为640x480,但它不工作。

python3 /yolov5 5/yolov5 5/tra.py-img-size 640 480 -批8-数据data.yaml - yolov5s.pt -缓存

代码语言:javascript
运行
复制
    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)

EN

回答 1

Stack Overflow用户

发布于 2022-11-26 08:09:37

代码语言:javascript
运行
复制
--img-size

只需要一个论点。使用:

代码语言:javascript
运行
复制
python3 /YOLOv5/yolov5/train.py --img-size 640 --batch 8 --epochs 300 --data data.yaml --weights yolov5s.pt --cache

图像的高度将相应调整,尊重高宽比和步幅需要。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/74457702

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档