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

使用YOLOv10训练自己的数据集

1. yolov10源码下载

THU-MIG/yolov10: YOLOv10: Real-Time End-to-End Object Detection (github.com)icon-default.png?t=N7T8https://github.com/THU-MIG/yolov10?tab=readme-ov-file

2. 环境配置

        预先安装好ANACONDA、PyCharm或者VSCode等基本软件。参考以下博客:

史上最全最详细的Anaconda安装教程-CSDN博客icon-default.png?t=N7T8https://blog.csdn.net/wq_ocean_/article/details/103889237(超详细)Python+PyCharm的安装步骤及PyCharm的使用(含快捷键)_python pycharm-CSDN博客icon-default.png?t=N7T8https://blog.csdn.net/junleon/article/details/120698578检测电脑里的python版本:

        我的是3.11。

使用Anaconda Prompt 创建虚拟环境,再激活环境。

conda create -n yolov10 python=3.11
conda activate yolov10 

安装支持包

pip install -r requirements.txt
pip install -e .

使用PyCharm打开下载的yolov10源码,右下角切换到创建的环境

创建train.py,执行以下代码

from ultralytics import YOLOv10
import osos.environ["GIT_PYTHON_REFRESH"] = "quiet"
os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"if __name__ == '__main__':model = YOLOv10('yolov10n.pt')result = model.train(data=coco128.yaml, epochs=40, batch=2, imgsz=640, device='cpu')

直接选用大家都有cpu训练,如果要使用GPU,将值改为0 即可,GPU使用依赖与yolov8一致。

YOLOv10 -Ultralytics YOLO 文档icon-default.png?t=N7T8https://docs.ultralytics.com/zh/models/yolov10/#model-variantsyolov10封装的很好,使用起来没什么难度,参考官方给的文档就能跑起来。

3. 使用所得权重文件预测图片

from ultralytics import YOLOv10
model = YOLOv10(r'C:\Users\Administrator\Desktop\yolov10-main\runs\detect\train4\weights\best.pt')
model.predict(save=True)

4. 性能分析

        总体效果还是比其他版本好很多的,但是我的训练时间貌似比v8长不少,这里论文中也给出了解释。

与其他最先进的探测器相比:

  • YOLOv10-S / X 比RT-DETR-R18 / R101 快 1.8 倍 / 1.3 倍,精度相似
  • 在精度相同的情况下,YOLOv10-B 比 YOLOv9-C 减少了 25% 的参数,延迟时间缩短了 46%
  • YOLOv10-L / X 的性能比YOLOv8-L / X 高 0.3 AP / 0.5 AP,参数少 1.8× / 2.3×

以下是 YOLOv10 变体与其他先进机型的详细比较:

模型参数 (M)FLOP (G)APval (%)延迟(毫秒)延迟(前向)(毫秒)
YOLOv6-3.0-N4.711.437.02.691.76
金色-YOLO-N5.612.139.62.921.82
YOLOv8-N3.28.737.36.161.77
YOLOv10-N2.36.739.51.841.79
YOLOv6-3.0-S18.545.344.33.422.35
金色-YOLO-S21.546.045.43.822.73
YOLOv8-S11.228.644.97.072.33
YOLOv10-S7.221.646.82.492.39
RT-DETR-R1820.060.046.54.584.49
YOLOv6-3.0-M34.985.849.15.634.56
金色-YOLO-M41.387.549.86.385.45
YOLOv8-M25.978.950.69.505.09
YOLOv10-M15.459.151.34.744.63
YOLOv6-3.0-L59.6150.751.89.027.90
金色-YOLO-L75.1151.751.810.659.78
YOLOv8-L43.7165.252.912.398.06
RT-DETR-R5042.0136.053.19.209.07
YOLOv10-L24.4120.353.47.287.21
YOLOv8-X68.2257.853.916.8612.83
RT-DETR-R10176.0259.054.313.7113.58
YOLOv10-X29.5160.454.410.7010.60

参考两位博主的论文分析

【论文阅读】 YOLOv10: Real-Time End-to-End Object Detection-CSDN博客icon-default.png?t=N7T8https://blog.csdn.net/shuaijieer/article/details/139182282

YOLOv10详细解读 | 一文带你深入了解yolov10的创新点(附网络结构图 + 举例说明)-CSDN博客icon-default.png?t=N7T8https://blog.csdn.net/java1314777/article/details/139206290

相关文章:

  • Java基础八股
  • opencv笔记(13)—— 停车场车位识别
  • 银行批量支付业务功能测试点详细介绍
  • C语言编程零基础:开启编程之旅的起点
  • 乡镇围栏数据存储ES实践
  • MySQL之创建高性能的索引(十二)
  • 设计模式(五)结构型模式---适配器模式
  • Unity DOTS技术(三)JobSystem+Burst+批处理
  • Web前端框架:深入探索与实践
  • 51单片机串口通信实现餐厅叫号
  • 数论1---整除
  • paddleocr快速入门:基于python脚本及命令行两种方式实现图片OCR识别
  • Qt 自定义代理类
  • ubuntu下搭建Supervisor
  • prometheus-alert使用
  • 2017-09-12 前端日报
  • Babel配置的不完全指南
  • canvas 高仿 Apple Watch 表盘
  • ERLANG 网工修炼笔记 ---- UDP
  • JavaScript新鲜事·第5期
  • java小心机(3)| 浅析finalize()
  • Kibana配置logstash,报表一体化
  • Linux快速配置 VIM 实现语法高亮 补全 缩进等功能
  • nodejs调试方法
  • python 学习笔记 - Queue Pipes,进程间通讯
  • Python进阶细节
  • Python连接Oracle
  • React的组件模式
  • Spring声明式事务管理之一:五大属性分析
  • Traffic-Sign Detection and Classification in the Wild 论文笔记
  • UMLCHINA 首席专家潘加宇鼎力推荐
  • Vue2.0 实现互斥
  • vue--为什么data属性必须是一个函数
  • win10下安装mysql5.7
  • 爱情 北京女病人
  • 警报:线上事故之CountDownLatch的威力
  • 异常机制详解
  • 因为阿里,他们成了“杭漂”
  • k8s使用glusterfs实现动态持久化存储
  • ​马来语翻译中文去哪比较好?
  • $refs 、$nextTic、动态组件、name的使用
  • (02)Hive SQL编译成MapReduce任务的过程
  • (07)Hive——窗口函数详解
  • (1)svelte 教程:hello world
  • (4)事件处理——(2)在页面加载的时候执行任务(Performing tasks on page load)...
  • (C++)八皇后问题
  • (cljs/run-at (JSVM. :browser) 搭建刚好可用的开发环境!)
  • (Mirage系列之二)VMware Horizon Mirage的经典用户用例及真实案例分析
  • (Note)C++中的继承方式
  • (第一天)包装对象、作用域、创建对象
  • (六)库存超卖案例实战——使用mysql分布式锁解决“超卖”问题
  • (原創) 是否该学PetShop将Model和BLL分开? (.NET) (N-Tier) (PetShop) (OO)
  • (转)大型网站架构演变和知识体系
  • (转)利用ant在Mac 下自动化打包签名Android程序
  • (转)人的集合论——移山之道