问题描述:

在使用yolov5训练自己模型,或者复现原文的数据时,一不小心程序断了,或者是使用服务器的时候断开连接了。这该怎么办?我也遇到了这,最开始是直接把【train.py】里面的这个参数

【–resume】的default设置成【True】,但是出现了下面的问题。这可把我愁坏了,还卡我好几天~~~

问题解决

经过一番请教大佬,得到了这个解决办法。给点赞!!!也是浅浅记录下。

首先,先找到你的训练模型文件,一般为【runs/train/exp/weights】目录下,该目录下会有这两个文件。

然后,按照下面的两个操作之一进行操作,就可以解决了!!!

将【train.py】里面的这个参数【–resume】的default设置成【runs/train/exp/weights/last.pt

在yolov5的根目录下,也就是【train.py】文件目录下,使用终端命令输入如下内容

python train.py --resume runs/train/exp/weights/last.pt

这样,问题就解决了,运行起来的效果如下。会接着你断开的程序继续运行~~~

到这,就已经解决了问题。欢迎关注蓝胖胖,咱们一起解决BUG~~~