首页 > 其他分享 >YOLO 实战演练

YOLO 实战演练

时间:2024-10-14 11:53:15浏览次数:5  
标签:实战 ultralytics 数据 YOLO mask 根目录 labelme data 演练

下面将会自定义训练目标检测数据集

注意:没有特殊说明,路径均是以项目根目录为准

官方数据集

首先下载权重文件并将放到 ultralytics 项目根目录

在这里插入图片描述

之后测试预训练模型的效果,在根目录执行如下命令

yolo predict model=yolov8n.pt source=ultralytics/assets/bus.jpg

在这里插入图片描述

之后我们查看保存的检测好的图片 /root/Development/ultralytics/runs/detect/predict2

在这里插入图片描述

训练 COCO128 数据集(这里可以配置 tensorboard 可视化面板,这里不赘述)

yolo train data=coco128.yaml model=yolov8n.pt epochs=3 lr0=0.01 batch=4

与此同时根目录下面生成了一个datasets文件夹,里面有 coco128 的数据集

注意:这里的 datasets 下载目录是在 ~/.config/Ultralytics/settings.yaml 文件中定义的,可以后续修改

在这里插入图片描述

之后查看存储的训练结果的文件夹,weights 文件夹里面装的是效果最好的一次权重文件以及最后一轮训练的权重文件

自定义数据集

下载数据集

【下载地址】:Mask Wearing Dataset - raw (roboflow.com)

在这里插入图片描述

然后下载 zip 压缩包即可

配置数据集

上传到项目根目录的 datasets,并重命名数据集为 MaskDataSet(本次演示直接使用下载的数据集,后面的部分步骤是针对自己制作数据集的要求)

data.yaml修改如下

path: …/datasets/MaskDataSet
train: ./train/images
val: ./valid/images
test: ./test/images

nc: 2
names: [‘mask’, ‘no-mask’]

roboflow:
workspace: joseph-nelson
project: mask-wearing
version: 4
license: Public Domain
url: https://universe.roboflow.com/joseph-nelson/mask-wearing/dataset/4

划分数据集

我们需要将数据集按照指定比例划分(训练集:验证集:测试集=7:2:1)

【数据集划分脚本】:division-of-data/DivisionOfData.py at main · kuisec/division-of-data (github.com)

标注数据集

安装 lableme,执行 pip install labelme,然后命令行输入 labelme即可进入图形化界面

但是注意:labelme 生成的标签是 json 文件的格式,后续需要转化成 txt 文件才能被 yolov 使用

这里说下两种标注工具 labelImg 和 labelme 的区别

labelimg 是一种矩形标注工具,常用于目标识别和目标检测,其标记数据输出为.xml和.txt

labelme 是一种多边形标注工具,可以准确的将轮廓标注出来,常用于分割,其标记输出格式为json

【数据集标注文件格式转换脚本】:DeepLearning/others/label_convert at master · KKKSQJ/DeepLearning (github.com)

训练数据集

yolo train data=datasets/MaskDataSet/data.yaml model=yolov8n.pt epochs=10 lr0=0.01 batch=4

在这里插入图片描述

使用效果最好的权重文件进行预测测试(需要提前上传 ultralytics/assets/mask.jpg口罩图片)

yolo predict model=runs/detect/train26/weights/best.pt source=ultralytics/assets/mask.jpg

查看最终效果

在这里插入图片描述

标签:实战,ultralytics,数据,YOLO,mask,根目录,labelme,data,演练
From: https://blog.csdn.net/m0_37302966/article/details/142912961

相关文章

  • 改进YOLOv8:通过注意力机制与模块优化实现高效目标检测【附保姆级代码】(YOLOv8)
    本专栏专为AI视觉领域的爱好者和从业者打造。涵盖分类、检测、分割、追踪等多项技术,带你从入门到精通!后续更有实战项目,助你轻松应对面试挑战!立即订阅,开启你的YOLOv8之旅!专栏订阅地址:https://blog.csdn.net/mrdeam/category_12804295.html文章目录改进YOLOv8:通过注意力......
  • 助力YOLOv8的突破—ODConv卷积技术的深度解析与实践【附保姆级代码】(YOLOv8)
    本专栏专为AI视觉领域的爱好者和从业者打造。涵盖分类、检测、分割、追踪等多项技术,带你从入门到精通!后续更有实战项目,助你轻松应对面试挑战!立即订阅,开启你的YOLOv8之旅!专栏订阅地址:https://blog.csdn.net/mrdeam/category_12804295.html文章目录助力YOLOv8的突破—ODC......
  • 毕业论文(设计)题目+毕设选题代码+paper+教程-计算机选题大集合-计算机视觉实战项目集
    选题(包括但不限于下列)基于YOLOv8的道路缺陷检测系统设计基于YOLOv5的手部关节骨龄预测系统设计基于CNN的雨雾天气下车辆检测和颜色识别系统基于YOLOv5算法的智慧零售商品识别系统设计基于YOLOv5的输电线路鸟巢检测系统设计基于深度学习的车型检测系统设计基于CNN的行......
  • yolov8+多算法多目标追踪+实例分割+目标检测+姿态估计(代码+教程)
    多目标追踪+实例分割+目标检测YOLO(YouOnlyLookOnce)是一个流行的目标检测算法,它能够在图像中准确地定位和识别多个物体。本项目是基于YOLO算法的目标跟踪系统,它将YOLO的目标检测功能与目标跟踪技术相结合,实现了实时的多目标跟踪。在目标追踪+语义分割+目标检......
  • 基于YOLOv8的道路缺陷检测系统设计(代码+教程+pyqt)
    路面裂缝检测是计算机视觉在基础设施维护中的一个重要应用。使用YOLOv8进行路面裂缝检测的过程与绝缘子检测类似,包括数据准备、模型训练和部署。下面是一个详细的流程,包括代码示例。YOLOv8是YouOnlyLookOnce(YOLO)系列目标检测算法的最新版本,它以其快速和准确的目标......
  • 垃圾堆积视频监测算法 YOLOv8
    垃圾堆积视频监测利用现场已有的监控摄像头,垃圾堆积视频监测通过人工智能AI视觉分析技术自动识别小区垃圾桶、街道、马路、路口是否有垃圾堆放、垃圾桶满溢等情况。一旦检测到垃圾堆放、垃圾桶满溢、垃圾暴露等情况,系统会自动截图并发出告警消息,提醒管理人员及时处理。与传统的......
  • yolov8中map指数提高
    引言:性能指标是评估对象检测模型的准确性和效率的关键工具。它们阐明了模型如何有效地识别和定位图像中的对象。这些见解对于评估和增强模型的性能至关重要。下面是我再yolov8训练过程中一些遇到的map过低的一些解决办法和见解  我遇到的问题是数据集过大,导致训练不充分......
  • 小红书私域引流实战打粉手册(2)
     前言大家好,我是渡鸦科技社的联合创始人:南叔,操盘全网最大的引流截流圈子,精通各行业的流量玩法,为社群兄弟解决从前端引流到后端出现的各种实操问题。南叔把实战引流的经验总结出来,免费分享给大家,希望对大家有帮助。废话不多说,直接上干货上期回顾➡️➡️➡️小红书私域引流实战......
  • 小红书私域引流实战打粉手册(3)
      前言大家好,我是渡鸦科技社的联合创始人:南叔,操盘全网最大的引流截流圈子,精通各行业的流量玩法,为社群兄弟解决从前端引流到后端出现的各种实操问题。南叔把实战引流的经验总结出来,免费分享给大家,希望对大家有帮助。废话不多说,直接上干货上期回顾➡️➡️➡️小红书私域引流实......
  • xml转YOLO的txt格式(一次必成版!)
    废话少说,直接上干货!首先,我先介绍一下代码的使用,编程语言为“Python”,共有三处需按照自己的需求修改的,我都放在代码最下面了。xml_root_path:输入你的xml格式的文件存放位置,建议全部用绝对路径txt_save_path:输入你的txt格式的文件导出后的存放位置classes_path:输入你的label......