保存训练权重
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仿真......希望能帮到你!

本文介绍了如何在YOLOv7中启用断点训练,包括修改train.py的resume参数,清理数据集缓存,以及在detect.py中调整iou_thres以避免检测框问题。作者还提供了进一步学习资源的相关链接。

1708

被折叠的 条评论
为什么被折叠?



