我们训练yolov5
代码时,一般会配置
一些参数,比如模型权重文件--weights
, 模型的配置文件--cfg
, 以及训练的数据--data
,
对应的训练脚本为:
训练train
python train.py -- weights './yolov5s.pt' --cfg 'models\yolov5s.yaml' --data './data/coco128.yaml'
Debug 参数设置
方法1: 直接代码中设置参数
那么对train.py 的代码进行Debug,如果不进行参数设置,直接Debug是会报错的。一种方法是手动在parse_opt
函数中修改
--weights , --cfg , --data
这三个参数,然后设置断点,按F5
进行调试。很显然这种方式需要手动去改代码,不是很方便,由于测试改动了参数下次重新改回来,很容易忘记原来的参数设置。
方法2: 在launch.json中配置参数
点击右边Debug
按钮, 选择创建launch.json
文件。
此时显示launch.json
的代码,如下所示:
{// 使用 IntelliSense 了解相关属性。 // 悬停以查看现有属性的描述。// 欲了解更多信息,请访问: https://go.microsoft.com/fwlink/?linkid=830387"version": "0.2.0","configurations": [{"name": "Python: Current File","type": "python","request": "launch","program": "${file}","console": "integratedTerminal","justMyCode": true}]
}
在launch.json
中,配置调试需要的参数, 新增一个args
变量,配置--weights
, --data
, --cfg
等需要配置的参数.
{"version": "0.2.0","configurations": [{"name": "Python: Current File","type": "python","request": "launch","program": "${file}","args": ["--weights", "./yolov5s.pt","--data", "./data/coco128.yaml","--cfg", "models\yolov5s.yaml",],"console": "integratedTerminal","justMyCode": true}]
}
这样就完成了训练参数的配置,就可以打断点,按F5
进行调试了,这个方式会比较方便点。