yolov5的训练中断恢复

Yolov5的恢复训练

方法一：使用自带参数-resume

1. train.py文件中找到函数parse_opt，修改参数–resume的默认参数为Ture

           

2. runs/train/exp*/weights/路径下找到上次训练的最后的保存的权重，加载断点之前最后保留的模型权重last.pt

             

3. 命令行输入命令训练：

python train.py --weights ./runs/train/exp20/weights/last.pt

在终端使用该命令时可以断点续练，但使用的是cpu，未查出是什么原因。

故使用方法二：更改parse_opt参数

1. train.py文件中找到函数parse_opt，修改参数–resume的默认参数为Ture

weights修改为last.pt  help=为文件地址，修改中断权重保存地址runs/train/exp 后直接运行即可

方法一参考链接：https://blog.csdn.net/CharmsLUO/article/details/123410081