前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >专栏 >YOLO11旋转目标识别(OBB)手把手教程: 如何训练自己的数据集(QR码为案列)

YOLO11旋转目标识别(OBB)手把手教程: 如何训练自己的数据集(QR码为案列)

原创
作者头像
AI小怪兽
发布于 2024-11-06 01:18:52
发布于 2024-11-06 01:18:52
3.2K00
代码可运行
举报
文章被收录于专栏:毕业设计毕业设计YOLO大作战
运行总次数:0
代码可运行

💡💡💡本文内容:YOLO11 OBB实现自有数据集缺陷旋转目标检测,从1)数据标记;2)数据json格式转换成适合yolo的txt格式;3)如何训练模型;

1.YOLO11介绍

Ultralytics YOLO11是一款尖端的、最先进的模型,它在之前YOLO版本成功的基础上进行了构建,并引入了新功能和改进,以进一步提升性能和灵活性。YOLO11设计快速、准确且易于使用,使其成为各种物体检测和跟踪、实例分割、图像分类以及姿态估计任务的绝佳选择。

OBB官方在 (DOTAv1)数据集上做了更多测试:

2. OBB旋转目标介绍

YOLO OBB格式通过四个角点指定边界框,坐标在0到1之间归一化。它遵循以下格式:

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
class_index, x1, y1, x2, y2, x3, y3, x4, y4

在内部,YOLO以xywhr格式处理损失和输出,xywhr格式表示边界框的中心点(xy)、宽度、高度和旋转。

2.1 labelme下载

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
# 安装labelme
pip install labelme

2.2使用labelme下

直接在python环境下运行

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
labelme

2.3 labelme介绍

1)Create Polygons生成polygon框;

3.QR码 旋转数据集介绍

训练集、验证集、测试集分别为:1894,100,101张

3.1 obb生成适合yolo格式的txt

obb_json_to_txt

YOLO11旋转目标识别(OBB)手把手教程: 数据集标注 | 数据格式转换 | 如何训练、测试-CSDN博客

4.OBB旋转目标训练

下载最新版即可,已支持OBB

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

4.1 qrcode-obb.yaml

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
path: D:/ultralytics-11/data/qrcode-obb/ 
train: train/images
val: valid/images
test: test/images

names: 
  0: qr_code

4.2 如何训练

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO('ultralytics/cfg/models/11/yolo11-obb.yaml')
    #model.load('yolov11.pt') # loading pretrain weights
    model.train(data='data/qrcode-obb.yaml',
                cache=False,
                imgsz=640,
                epochs=200,
                batch=16,
                close_mosaic=10,
                device='0',
                optimizer='SGD', # using SGD
                project='runs/train',
                name='exp',
                )

4.3训练结果可视化

代码语言:javascript
代码运行次数:0
运行
AI代码解释
复制
YOLO11-obb summary (fused): 300 layers, 2,897,630 parameters, 0 gradients, 6.6 GFLOPs
                 Class     Images  Instances      Box(P          R      mAP50  mAP50-95): 100%|██████████| 7/7 [00:04<00:00,  1.53it/s]
                   all         99        128      0.953      0.955      0.966      0.908

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
暂无评论
推荐阅读
编辑精选文章
换一批
YOLO11-pose关键点检测:训练实战篇 | 自己数据集从labelme标注到生成yolo格式的关键点数据以及训练教程
💡💡💡本文解决什么问题:教会你如何用自己的数据集转换成对应格式的数据集以及如何训练YOLO11-pose关键点检测
AI小怪兽
2024/10/17
4.1K0
YOLO11-seg分割如何训练自己的数据集(道路缺陷)
💡💡💡本文内容:如何用自己的数据集(道路缺陷)训练yolo11-seg模型以及训练结果可视化;
AI小怪兽
2024/10/21
1.3K0
YOLOv8 OBB实现自有数据集缺陷旋转检测,从数据标记格式转换到训练的手把手教程
💡💡💡本文内容:YOLOv8 OBB实现自有数据集缺陷旋转检测,从数据标记到训练的手把手教程
AI小怪兽
2024/01/20
7.5K3
基于YOLO11的铁路轨道缺陷检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的铁路轨道缺陷检测,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/03/21
1731
YOLO11-seg分割:如何训练自己的数据集:包裹分割数据集
💡💡💡本文内容:如何训练包裹分割数据集,包装分割数据集(Package Segmentation Dataset)推动的包装分割对于优化物流、加强最后一英里配送、改进制造质量控制以及促进智能城市解决方案至关重要
AI小怪兽
2024/12/14
8430
基于YOLO11的焊膏不良缺陷检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的焊膏不良缺陷检测系统,阐述了整个数据制作和训练可视化过程
AI小怪兽
2024/12/30
2180
基于YOLO11的作物与杂草检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的作物与杂草检测,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/04/01
1820
基于YOLO11的水下物体检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的水下物体检测,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/03/28
2550
YOLO11教程:如何训练 | 验证 | 测试 自己的数据集源码分享
Ultralytics YOLO11是一款尖端的、最先进的模型,它在之前YOLO版本成功的基础上进行了构建,并引入了新功能和改进,以进一步提升性能和灵活性。YOLO11设计快速、准确且易于使用,使其成为各种物体检测和跟踪、实例分割、图像分类以及姿态估计任务的绝佳选择。
AI小怪兽
2024/10/30
5.3K0
基于YOLO11的交通标志(traffic-sign)检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的交通标志检测,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/03/15
2740
基于YOLO11的茶叶嫩芽检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的茶叶嫩芽检测,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/03/25
2110
基于YOLO11的半导体晶圆缺陷检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的半导体晶圆缺陷检测,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/03/11
2012
YOLO11实战:GC10-DET缺陷检测 | DCNv4结合SPPF+11Detect创新性结合,创新十足
💡💡💡本文独家改进:DCNv4更快收敛、更高速度、更高性能,完美和YOLO11结合,助力涨点
AI小怪兽
2024/12/29
3160
基于YOLOv8的晶体管定位识别(pose),从数据集标注到训练手把手教程
💡💡💡本文解决什么问题:教会你如何用晶体管从标注到训练Yolov8-pose关键点检测
AI小怪兽
2024/01/22
2K4
YOLO11-seg分割:SPPF_attention,重新设计加入注意力机制,助力分割
优点:为了利用不同的池化核尺寸提取特征的方式可以获得更多的特征信息,提高网络的识别精度。
AI小怪兽
2024/10/23
1.1K0
基于YOLO11的无人机检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的无人机检测,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/03/16
2080
基于YOLO11的车牌(CCPD2020)检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的车牌(CCPD2020)检测,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/03/17
2990
基于YOLO11的番茄检测系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的番茄检测,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/03/26
3540
Yolo11实战改进:一文读懂Yolo11到实战
Yolo11是Ultralytics推出的新一代计算机视觉模型,为YOLO家族树立了新的里程碑。它在对象检测、实例分割、图像分类、姿势估计、定向物体检测和对象跟踪等计算机视觉任务上展现了卓越的性能和准确性,相较于YOLOv8有了显著提升。
AI浩
2024/11/18
3.2K0
Yolo11实战改进:一文读懂Yolo11到实战
基于YOLO11的汽车损伤分割系统(Python源码+数据集+Pyside6界面)
💡💡💡本文摘要:基于YOLO11的汽车损伤分割,阐述了整个数据制作和训练可视化过程
AI小怪兽
2025/03/27
1760
推荐阅读
YOLO11-pose关键点检测:训练实战篇 | 自己数据集从labelme标注到生成yolo格式的关键点数据以及训练教程
4.1K0
YOLO11-seg分割如何训练自己的数据集(道路缺陷)
1.3K0
YOLOv8 OBB实现自有数据集缺陷旋转检测,从数据标记格式转换到训练的手把手教程
7.5K3
基于YOLO11的铁路轨道缺陷检测系统(Python源码+数据集+Pyside6界面)
1731
YOLO11-seg分割:如何训练自己的数据集:包裹分割数据集
8430
基于YOLO11的焊膏不良缺陷检测系统(Python源码+数据集+Pyside6界面)
2180
基于YOLO11的作物与杂草检测系统(Python源码+数据集+Pyside6界面)
1820
基于YOLO11的水下物体检测系统(Python源码+数据集+Pyside6界面)
2550
YOLO11教程:如何训练 | 验证 | 测试 自己的数据集源码分享
5.3K0
基于YOLO11的交通标志(traffic-sign)检测系统(Python源码+数据集+Pyside6界面)
2740
基于YOLO11的茶叶嫩芽检测系统(Python源码+数据集+Pyside6界面)
2110
基于YOLO11的半导体晶圆缺陷检测系统(Python源码+数据集+Pyside6界面)
2012
YOLO11实战:GC10-DET缺陷检测 | DCNv4结合SPPF+11Detect创新性结合,创新十足
3160
基于YOLOv8的晶体管定位识别(pose),从数据集标注到训练手把手教程
2K4
YOLO11-seg分割:SPPF_attention,重新设计加入注意力机制,助力分割
1.1K0
基于YOLO11的无人机检测系统(Python源码+数据集+Pyside6界面)
2080
基于YOLO11的车牌(CCPD2020)检测系统(Python源码+数据集+Pyside6界面)
2990
基于YOLO11的番茄检测系统(Python源码+数据集+Pyside6界面)
3540
Yolo11实战改进:一文读懂Yolo11到实战
3.2K0
基于YOLO11的汽车损伤分割系统(Python源码+数据集+Pyside6界面)
1760
相关推荐
YOLO11-pose关键点检测:训练实战篇 | 自己数据集从labelme标注到生成yolo格式的关键点数据以及训练教程
更多 >
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档
本文部分代码块支持一键运行,欢迎体验
本文部分代码块支持一键运行,欢迎体验