【目标检测|yolov5——断点训练/继续训练【解决方法、使用教程】】
yolov5——断点训练/继续训练【解决方法、使用教程】
- 1. 问题描述
- 2. 解决办法
-
- 2.1 步骤一
- 2.2 步骤二
- 2.3 步骤三
- 3. 全部流程
1. 问题描述 在用yolov5训练数据的过程中由于突发情况训练过程突然中断，从头训练耗时，想接着上次训练继续训练怎么办，教你一招搞定，在yolov5中给我们提供现成的参数–resume。
文章图片
2. 解决办法 2.1 步骤一 再train.py文件中找到函数parse_opt，修改参数–resume的默认参数为Ture
文章图片
2.2 步骤二 再runs/train/exp*/weights/路径下找到上次训练的最后的保存的权重，加载断点之前最后保留的模型即可，例如笔者的
文章图片
2.3 步骤三 命令行输入命令：
文章图片
3. 全部流程
文章图片
推荐阅读
- 激光条纹中心提取|opencv二值图像分割——python
- 目标检测|yolov5——detect.py代码【注释、详解、使用教程】
- 计算机视觉|平均池化和最大池化分别适用于什么场景呢（）
- 学习笔记|如何使用PyCharm对函数进行测试
- 接口自动化测试框架|接口自动化--数据库
- 接口自动化测试框架|接口自动化--日志管理
- python自动化测试|python自动化之数据驱动 - data driven（参数化）
- python|分享5个你不知道的软件测试小神器（赶紧收藏起来）
- chrome|Selenium Chrome驱动安装(windows系统)