当前位置: 首页 > article >正文

yolo继续训练模型

使用YOLO训练好自己的模型并持续训练

    • 一、断点续传
    • 二、在上次训练结果的基础上继续开始训练
    • 三、last.pt 和 best.pt区别

运行环境:Ubuntu18.04 + anaconda虚拟环境

一、断点续传

对于上次训练时意外中断的情况

将patser中第9个参数resume,设置为default=True

或运行:

python train.py --resume True

二、在上次训练结果的基础上继续开始训练

将patser中第1个参数weights,设置为上次运行完

~/yolov5/runs/train/exp/weights/last.pt

同样可以在运行时进行修改,如:

python train.py --weights runs/train/exp/weights/last.pt

三、last.pt 和 best.pt区别

有博主提示:不能加载best,只能加载last,best里面没有节点信息。
last.pt 和 best.pt 是在 YOLO 训练过程中生成的两个不同的模型权重文件,它们之间的主要区别如下:

  1. last.pt

定义:这是训练过程中最后一个 epoch 的模型权重,表示训练结束时的状态。

包含所有训练迭代的学习成果。
不一定是性能最好的模型,可能在训练后期过拟合于训练数据。

  1. best.pt

定义:这是在训练过程中根据验证集性能(如 mAP 或损失值)保存的最佳模型权重。

在训练过程中,模型在验证集上的表现最好的时刻的权重。
通常是在多个 epoch 中保存的最佳结果,旨在提高模型的泛化能力。

  1. 对pt文件的选择使用
  • 使用 last.pt:如果你希望在模型训练的最后状态基础上进行进一步训练或评估。
  • 使用 best.pt:如果你想要在验证集上表现最佳的模型进行推理或部署。

http://www.kler.cn/a/383584.html

相关文章:

  • RAGFlow 基于深度文档理解构建的开源 RAG引擎 - 安装部署
  • Mysql 查询性能调优总结
  • RPA系列-uipath 学习笔记3
  • AIA - IMSIC之二(附IMSIC处理流程图)
  • 如何在谷歌浏览器中进行网络速度测试
  • 聊天社交管理系统 Java 源码,构建个性化社交空间
  • 【Linux内存泄漏】自创pamp 内存快照比对定位内存泄漏【2024-11-07】
  • npm镜像的常用操作
  • 职场逆袭!学会管理上司,你也能成为职场赢家
  • C语言 | Leetcode C语言题解之第524题通过删除字母匹配到字典里最长单词
  • 代码随想录算法训练营第二十一天 | LeetCode93.复原IP地址、LeetCode78.子集、LeetCode90.子集II
  • RFID应急消防管控:科技与效率的完美结合
  • golang学习2
  • 轮播图【HTML+CSS+JavaScript】
  • ubuntu 之 压缩与解压缩(7zip,zip,tar.gz,rar...)
  • 从零开始学python 6(持续更新中ing)
  • 知识总结三
  • Webserver(4.3)TCP通信实现
  • 基于CNN-BiLSTM的时间序列数据预测,15个输入1个输出,可以更改数据集,MATLAB代码
  • V4L2 sub-devices 翻译
  • Python基础学习_01
  • Android 使用自定义注解标注当前类
  • STM32学习笔记-外部中断和外部时钟
  • 前端学习笔记—Vue3特性
  • web安全测试渗透案例知识点总结(下)——小白入狱
  • Zookeeper分布式锁实现