yolov5训练模型命令行
时间: 2023-10-14 18:08:02 浏览: 208
YOLOV5转换模型为om模型文件脚本
yolov5训练模型的命令行如下:
```
python train.py --img 640 --batch 16 --epochs 50 --data coco.yaml --cfg models/yolov5s.yaml --weights '' --name yolov5s_results
```
其中,`--img`指定输入图像的大小,`--batch`指定批量大小,`--epochs`指定训练轮数,`--data`指定数据集的配置文件,`--cfg`指定模型的配置文件,`--weights`指定预训练模型的权重文件,`--name`指定训练结果的保存路径。
阅读全文