0 YOLO系列笔记
1 前言
本文汇总在使用yolo过程中遇到的一些问题,时时更新。
2 训练中断怎么办?
如果自己的电脑性能不够,导致训练过程中蓝屏了;或者是因为什么必要的原因必须关掉电脑,那么就会面临训练了很久的YOLO模型被迫中断,好在YOLOv5中已经集成了断点续传 的功能,遇到这种情况,完全不用担心。
解决办法就是稍微改动一下train.py的源码,将--resume
参数的默认值改为True
,然后再运行之前训练数据集的指令即可。它会直接检测上一次的遗留,然后接着断点继续训练。
注意:如果直接在命令行参数传入
--resume
参数的值,经过测试,发现会报错,可能是什么代码没有写好。