YOLOV8训练中断之后如何继续上次训练？

已于 2023-06-02 10:48:12 修改
阅读量3.9k 收藏 16
点赞数 7
文章标签： YOLO 深度学习 python
于 2023-06-01 20:37:19 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_44024748/article/details/130995637
版权

该方法尝试成功，但是会新创建一个文件夹存放继续训练的权重文件等信息，而且接着训练P R精度并没有接着上次中断的继续更新，这一点有些奇怪还没有找到原因

第一步： 打开ultralytics/yolo/engine/model.py代码

                找到def train() 函数，将self.trainer.model = self.model注释掉

 第二步：打开ultralytics/yolo/engine/trainer.py代码

                首先，找到check_resume()函数, 将resume = self.args.resume注释掉，再让resume = 上次中断生成的last.pt的路径

                 其次，找到resume_training()函数，ckpt =torch.load(上次中断生成的last.pt的路径) 

 切记，下次从头开始训练时，一定要将代码恢复回原样！！！！

该方法YOLOV8不可行，或许YOLOV5可行，YOLOV5可以直接在配置文件中将resume设置为True：Yolov8断点续训/继续训练_q1224352995的博客-CSDN博客

 

weixin_44024748
关注
  • 7
    点赞
  • 踩
  • 16
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
YOLOV8多类别车辆检测训练好的模型+数据集
04-23
1、YOLOV8多类别车辆检测训练好的模型和YOLO多类别车辆检测数据集，类别为ambulance、bus、truck， car， Motorcycle五个类别，标签格式我txt和xml两种，已经划分好train、val、test、可以直接用于YOLO车辆检测等等...
Yolov8如何在训练意外中断后接续训练
qq_43471945的博客
03-30 1万+
在训练YOLOv8的时候，因为开太多其他程序，导致在100多次的时候崩溃，查询网上相关知识如何接着训练，在yolo5中把resume改成True就可以。在yolov8中也这样尝试，将ultralytics/yolo/cfg/default.yaml中的resume改成True发现并没有作用，感觉yolov8代码还是有很多bug。
YOLOv8从断点处继续训练
如烟
03-01 6272
找到上次训练保存的路径位置（我的是：/home/zqq/下载/ultralytics-main/ultralytics/yolo/v8/detect/runs/detect/train10/weights）顺便说个小技巧，ctrl+c是中止任务，ctrl+z是将任务放在后台运行，可以用fg命令重新调回前台。我的电脑是笔记本版本的3060（6G显存），batchsize设了16，epoch设了300。第一次用YOLOv8，发现代码结构跟以前的差别挺大的，看了文档才知道大概用法。
Yolov8中断后继续训练，resume方法，断点训练
最新发布
weixin_61623830的博客
05-21 1361
1.错误尝试在训练YOLOv8的时候，因为开太多其他程序，导致在100多次的时候崩溃，查询网上相关知识如何接着训练，在yolo5中把resume改成True就可以。在yolov8中也这样尝试，将ultralytics/yolo/cfg/default.yaml中的resume改成True发现并没有作用，感觉yolov8代码还是有很多bug。
YOLOv8继续训练
weixin_61814684的博客
10-25 888
进入对应环境和YOLOv8主文件夹下，输入如下命令即可（path/to/last.pt替换成自己训练的last.pt位置，一般是以run开头，例如runs/detect/train7/weights/last.pt）
YOLOv8常见问题解决方案
AsUs.的博客
02-23 1118
通过设置 resume=True，"…Ultralytics YOLO 您可以通过设置 resume 参数 True 在调用 train 方法的路径，并指定 .pt 文件，其中包含经过部分训练的模型权重。恢复训练时，Ultralytics YOLO 会加载上次保存模型的权重，并恢复优化器状态、学习率调度器和历时编号。这样，您就可以从上次中断的地方无缝地继续训练过程。请记住，默认情况下，检查点会在每个纪元结束时保存，或者使用 save_period 参数，因此您必须至少完成一个纪元才能恢复训练运行。
YOLOv8训练好模型后，追加轮数继续训练、或者提前终止训练，缩减训练轮数
kiong_的博客
03-17 5435
而且此教程适用的情况是你已经训练好了此模型，想继续追加一些轮数。比如训练进度是120/120，已经完成了，继续追加10轮，或者你原先定的是200轮，希望缩减到150轮，可以使用我说的这个方法。为什么缩减也要改一些东西呢？我直接不训练了不就可以了吗？这是因为没有训练完所有epoch，不会输出P、PR曲线等指标；或者说你不想一直盯着看，希望准确地缩减到某轮数，自动停止。如果你不需要追加训练，只想缩减训练，那么直接看这个博客即可：YOLOv8断点恢复、减少训练轮数、提前终止。这篇博客是本篇博客的一个子集。如果想追
Yolov8中断后继续训练
su_xiao_wei的专栏
12-05 2644
Yolov8中断后继续训练/断点训练
YOLOv8如何进行断点续训？
毕竟是Shy哥
10-30 1483
【代码】YOLOv8如何进行断点续训？
YOLOV8进行resume接续训练
qq_69854365的博客
10-12 2384
路径在ultralytics/yolo/engine/trainer.py。路径在ultralytics/yolo/engine/trainer.py。路径在ultralytics/yolo/engine/model.py中。修改train中模型的加载方式。01找到check_resume方法。03修改完整体的代码为。03替换为直接加载路径。03修改完整体的代码为。02 注释掉下面代码。
yolov8训练中断后重新开始训练
Cr_NanMao的博客
04-27 4694
将cfg中的default.yaml配置文件中的resume参数改成True。将engine中的model.py的self.model = self.trainer.model这行代码注释掉。 第三件事： 在engine中的trainer.py做以下修改1. 将check_resume方法中的 resume = self.args.resume 改成 resume = r'G:\yolov8\ultralytics\yolo\v8\detect\runs\detect\train2\w
yolov8断点恢复训练及减少训练次数和增加训练次数
热门推荐
darkredrock的博客
06-05 1万+
在ultralytics/yolo/engine/trainer.py中找到self.epochs = self.args.epochs，把self.epochs 给成自己需要的次数。第二种情况是没有安装上诉两个包，或者不用上诉两个包训练而使用python脚本训练。注释check_resume中resume = self.args.resume，改成需要断点恢复的last.pt。如果需要增加训练次数把self.epochs改成自己需要的次数，比如以下是200次恢复训练，需要提高到300次。
yolov8训练自己的数据集（源码）.rar
05-09
让我们深入了解一下YOLOv8及其训练自定义数据集的关键知识点。 1. YOLO（You Only Look Once）系列： YOLO是一种实时目标检测系统，由Joseph Redmon等人在2016年首次提出。YOLOv8是该系列的最新版本，它在前几代的...
YOLOv8训练飞机检测模型+权重+数据集
04-18
1、YOLOv8训练飞机检测模型，并包含标注好的飞机检测数据集，标签格式为xml和txt两种，类别名为aeroplane， 2、数据集和检测结果参考：https://blog.csdn.net/zhiqingAI/article/details/124230743 5、采用pytrch...
YOLOv8训练好的火焰烟雾检测模型+数据集
04-25
1、YOLOv8训练好的火焰烟雾检测模型，并包含标注好的火焰和烟雾数据集，标签格式为txt两种，类别名为fire和smoke， 2、数据集和检测结果参考：https://blog.csdn.net/zhiqingAI/article/details/124230743 4、采用...
YOLOv8预训练权重文件集合(YOLOv8n,YOLOv8s,YOLOv8m,YOLOv8l,YOLOv8x)
06-23
YOLOv8预训练权重文件集合(YOLOv8n,YOLOv8s,YOLOv8m,YOLOv8l,YOLOv8x) YOLOv8 pretrained Detect models are shown here. Detect, Segment and Pose models are pretrained on the COCO dataset, while Classify ...
Yolov8断点续训/继续训练
q1224352995的博客
05-02 4567
我看好多博主说的是用命令行进行续训，还有一位博主改了trainer.py文件和model.py文件的。因为YOLOv8显存一直变化的原因我的挂的程序也被挤下来，这些方法不适合我这种不用命令行启动程序的人。一开始我把default.py文件中的resume行设为true，model设为我想要的路径，但它老会跑到另一个正在运行的YOLOv8副本文件夹中的权重进行继续训练，后面查看了train.py中的代码。分析这个功能发现只要将文件路径直接写到resume后面就行了。不用写true直接填自己路径即可。
yolov8断点续训教程
我不想CV的博客
10-10 580
yolov8断点续续
yolov8断点继续训练
08-24
您好！对于YOLOv8模型的断点继续训练，您需要执行以下步骤： 1. 首先，确保您已经安装了YOLOv8的相关依赖和环境。 2. 在进行断点继续训练之前，您需要先下载预训练的权重文件。这些预训练的权重文件包含了已经学到的特征参数，能够为模型提供初始化的参数。 3. 接下来，您需要修改YOLOv8的配置文件，将训练设置中的相关参数进行调整。主要需要修改的是`train_from`参数，将其设置为上次训练的权重文件路径。 4. 然后，您可以运行训练脚本以进行断点继续训练。根据您使用的具体框架和代码库不同，具体的脚本命令可能会有所不同。通常情况下，您需要指定训练数据集、配置文件、权重文件等参数。 5. 在断点继续训练期间，系统会加载上一次训练保存的权重文件，并在此基础上继续优化模型。您可以通过设置适当的训练轮数或其他相关参数来控制训练的持续时间。 请注意，断点继续训练可能需要较长的时间和较高的计算资源，具体取决于您的数据集大小和模型复杂度。同时，确保您的训练数据集和标签文件与上次训练时的一致，以保证断点继续训练的正确性。 希望以上信息对您有所帮助！如果您有任何其他问题，请随时提问。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值