保存训练权重
YOLOv7默认保存第0轮训练结果以及最后5轮训练结果,中间训练结果每隔25轮默认保存一次;YOLOv5默认保存最优模型以及最后模型。保存最后一轮模型的优点为:为防止训练过程中断开训练,可以使用 --resume 命令在最后一轮的基础上进行训练。
断点训练
①更改train.py文件的参数
将上图中的 –resume 参数的 default=False 设置为 True。这一步是将解释器中的断点续训设置为True(即进行断点续训),如果是从头开始训练就不需要更改这个参数值。
②清理数据集中的*.cache缓存
在之前的训练中会出现*.cache缓存,如果不清理的话,YOLOv7断点续训会失败(YOLOv7在每次训练开始的时候都要清除上一次训练的数据集缓存,不然都会出现训练失败的情况)。清理完缓存,在断点续训开始的时候,会重新生成数据集索引。
detect.py
在detect.py中,iou_thres设得过大,会出现检测框重叠得情况;若值设得过小,会发生误检。
最后:
如果你想要进一步了解更多的相关知识,可以关注下面公众号联系~会不定期发布相关设计内容包括但不限于如下内容:信号处理、通信仿真、算法设计、matlab appdesigner,gui设计、simulink仿真......希望能帮到你!