YOLOv5resume训练
需求:YOLOv5已经训练30轮,觉得还未好的效果,想继续训练200轮。
问题一、AssertionError: runs\train\yolov5l-C3-VOC07+12-baseline\weights\last.pt training to 50 epochs is finished, nothing to resume.

修改了代码这两处:

解决:
step1:
先将–resume的defualt改为True,再将–epochs设为想要继续训练到的epochs,即defualt=200

step2:
在trainds.py文件中,按下Ctrl+F搜索Epochs定位,做以下两处修改:
添加一行代码:ckpt[‘epoch’] = 30
修改代码:start_epoch = ckpt[‘epoch’](后面不加1)
如图:

当YOLOv5在30轮训练后未达到理想效果,欲继续训练至200轮时遇到AssertionError。解决方法包括修改–resume的默认值为True,设置–epochs为200,并在trainds.py中调整Epochs相关代码,如设置ckpt['epoch'] = 30及修改start_epoch的计算。参考链接提供了详细步骤。
最低0.47元/天 解锁文章
10万+





