train.py文件在pycharm中输入参数命令的方法

保姆式yolov5教程,训练你自己的数据集 - 知乎 (zhihu.com)

点击train -> Edit Configurations:

 

在Parameters,输入对应参数命令

--weights yolov5s.pt --data data/fire.yaml --workers 1 --batch-size 8 

假如没有输入参数的界面,则对代码设置断点调试一下,就有了。

posted on 2022-08-20 12:31  一杯明月  阅读(212)  评论(0)    收藏  举报