问题描述:
在使用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~~~