YOLOv5resume训练

YOLOv5resume训练

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

需求:YOLOv5已经训练30轮,觉得还未好的效果,想继续训练200轮。

问题一、AssertionError: runs\train\yolov5l-C3-VOC07+12-baseline\weights\last.pt training to 50 epochs is finished, nothing to resume.

YOLOv5resume训练_第1张图片
修改了代码这两处:
YOLOv5resume训练_第2张图片

解决:

step1:
先将–resume的defualt改为True,再将–epochs设为想要继续训练到的epochs,即defualt=200
在这里插入图片描述

step2:
在trainds.py文件中,按下Ctrl+F搜索Epochs定位,做以下两处修改:

添加一行代码:ckpt[‘epoch’] = 30
修改代码:start_epoch = ckpt[‘epoch’](后面不加1)
如图:
YOLOv5resume训练_第3张图片

step3:
YOLOv5resume训练_第4张图片
YOLOv5resume训练_第5张图片
解决参考:https://blog.csdn.net/weixin_42886071/article/details/126676646

你可能感兴趣的:(深度学习)