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

【目标检测】使用自己的数据集训练并预测yolov8模型

1、下载yolov8的官方代码

地址: GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

2、下载目标检测的训练权重   yolov8n.pt

                        将  yolov8n.pt 放在ultralytics文件夹下

3、数据集分布

             注意(测试集可有可无)

     ----mydata

           -------train

                   --------images    (训练图片)

                   --------labels      (训练标签)

           -------test

                   --------images   (测试图片)

                   --------labels     (测试标签)

           -------valid

                   --------images   (验证图片)

                   --------labels     (验证标签)

 4、修改文件的参数

   (1) 修改coco8.yaml的参数

     将....../ultralytics/ultralytics/cfg/datasets 目录下的 coco8.yaml复制到ultralytics中去。

修改里面的参数,主要是数据集的路径和类别信息。

     

      (2)修改yolov8.yaml的参数

文件位置在....../ultralytics/ultralytics/cfg/models/v8/yolov8.yaml

nc: 代表yolo8有80个类别,这里改为我们自己数据集的类别数

   (3) 卸载ultralytics ,然后在重新装一下

 卸载:

pip uninstall ultralytics

安装

python  setup.py

最新版本的ultralytics是没有setup.py这个文件的,所以安装方式参考以下博客

快速教你解决yolov8没有setup.py的问题_没有setup.py文件-CSDN博客

5、训练自己的数据集

yolo task=detect mode=train model=/home/dyy/PycharmProjects/ultralytics/ultralytics/cfg/models/v8/yolov8n.yaml data=/home/dyy/PycharmProjects/ultralytics/tank.yaml(自己的数据集参数) epochs=10 batch=1 save=true workers=8 

6、测试自己的数据集

yolo task=detect mode=predit model=/home/dyy/PycharmProjects/ultralytics/runs/detect/train12/weights/best.pt source=/home/dyy/PycharmProjects/ultralytics/myData/color/355.png save=true

相关文章:

  • 北京网站建设多少钱?
  • 辽宁网页制作哪家好_网站建设
  • 高端品牌网站建设_汉中网站制作
  • 行列视(RCV)是否支持批量共享和自定义共享设置,以提高共享效率?
  • hive 调优
  • 科技日报社激发数据要素价值,树立媒体行业数字化转型标杆
  • Java--多态
  • 前端调试技巧(npm Link,vscode调试,浏览器调试等)
  • C++入门知识
  • 全球激光位移传感器市场规模逐渐扩大 企业数量不断增多
  • 守护舌尖安全,破解EHS管理难题,食品加工企业的可持续发展之路
  • Vatee万腾平台:创新科技,驱动未来
  • 关于 Node.js、Deno 和 Bun 的思考
  • 企业如何做好数据防泄密丨十个措施杜绝泄密事件
  • android13 固定U盘链接 SD卡链接 TF卡链接 硬盘链接
  • 为什么要参加学术会议?
  • 油烟监测仪:守护厨房,让蓝天白云成为常态
  • EasyOCR: 简单易用的多语言OCR工具
  • 【Redis学习笔记】2018-06-28 redis命令源码学习1
  • css选择器
  • exif信息对照
  • interface和setter,getter
  • Java超时控制的实现
  • JS 面试题总结
  • Making An Indicator With Pure CSS
  • ng6--错误信息小结(持续更新)
  • PHP变量
  • Python 反序列化安全问题(二)
  • 坑!为什么View.startAnimation不起作用?
  • 马上搞懂 GeoJSON
  • 适配iPhoneX、iPhoneXs、iPhoneXs Max、iPhoneXr 屏幕尺寸及安全区域
  • 与 ConTeXt MkIV 官方文档的接驳
  • raise 与 raise ... from 的区别
  • 微龛半导体获数千万Pre-A轮融资,投资方为国中创投 ...
  • ​​​​​​​开发面试“八股文”:助力还是阻力?
  • ​【数据结构与算法】冒泡排序:简单易懂的排序算法解析
  • ‌JavaScript 数据类型转换
  • # Kafka_深入探秘者(2):kafka 生产者
  • ###C语言程序设计-----C语言学习(6)#
  • #我与Java虚拟机的故事#连载04:一本让自己没面子的书
  • (STM32笔记)九、RCC时钟树与时钟 第一部分
  • (八)Docker网络跨主机通讯vxlan和vlan
  • (附源码)springboot教学评价 毕业设计 641310
  • (附源码)ssm教师工作量核算统计系统 毕业设计 162307
  • (附源码)计算机毕业设计ssm-Java网名推荐系统
  • (企业 / 公司项目)前端使用pingyin-pro将汉字转成拼音
  • (全注解开发)学习Spring-MVC的第三天
  • (学习日记)2024.04.10:UCOSIII第三十八节:事件实验
  • (一)Linux+Windows下安装ffmpeg
  • (转)Oracle存储过程编写经验和优化措施
  • ***详解账号泄露:全球约1亿用户已泄露
  • **《Linux/Unix系统编程手册》读书笔记24章**
  • .NET 4.0中的泛型协变和反变
  • .NET CF命令行调试器MDbg入门(二) 设备模拟器
  • .net CHARTING图表控件下载地址
  • .NET gRPC 和RESTful简单对比
  • .NET WebClient 类下载部分文件会错误?可能是解压缩的锅
  • .Net 高效开发之不可错过的实用工具