Yolov13论文数据公开:实验原始数据与代码复现指南

Yolov13论文数据公开:实验原始数据与代码复现指南

【免费下载链接】Yolov13 【免费下载链接】Yolov13 项目地址: https://ai.gitcode.com/hf_mirrors/atalaydenknalbant/Yolov13

项目概述

Yolov13是一款基于超图增强自适应视觉感知的实时目标检测模型,通过HyperACE机制和FullPAD范式实现了更高的检测精度和效率。本指南将详细介绍如何获取实验原始数据并复现论文中的关键实验结果。项目提供了四个模型变体:Nano(yolov13n.pt)、Small(yolov13s.pt)、Large(yolov13l.pt)和X-Large(yolov13x.pt),满足不同场景的需求。

技术背景

HyperACE机制

Hypergraph-based Adaptive Correlation Enhancement(超图自适应关联增强)机制将多尺度特征图中的像素视为超图顶点,通过可学习的超边构建模块自适应探索顶点间的高阶关联。该机制采用线性复杂度的消息传递模块,在高阶关联的引导下有效聚合多尺度特征,实现复杂场景的有效视觉感知。

FullPAD范式

Full-Pipeline Aggregation-and-Distribution(全流程聚合与分发)范式利用HyperACE聚合骨干网络的多尺度特征,并在超图空间中提取高阶关联。通过三个独立通道将关联增强特征分别传递到骨干与颈部连接、颈部内部层以及颈部与头部连接,实现全流程的细粒度信息流和表示协同。

实验数据获取

模型权重下载

项目提供的预训练模型权重文件位于当前目录下,包括:

MS COCO数据集

实验使用的MS COCO数据集需用户自行准备,数据集配置文件可参考官方文档中的说明。

环境搭建

依赖安装

# 创建虚拟环境
conda create -n yolov13 python=3.11
conda activate yolov13

# 安装依赖
pip install -r requirements.txt
pip install -e .

# 安装Flash Attention加速库
wget https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
pip install flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl

实验复现步骤

模型验证

使用以下代码在COCO数据集上验证模型性能:

from ultralytics import YOLO

# 加载模型,替换{n/s/l/x}选择不同版本
model = YOLO('yolov13n.pt')

# 在验证集上评估模型
metrics = model.val()
print(metrics.box.map)  # 打印mAP50-95
print(metrics.box.map50)  # 打印mAP50
print(metrics.box.map75)  # 打印mAP75

训练复现

以下是复现论文中训练结果的代码示例:

from ultralytics import YOLO

# 加载模型配置文件
model = YOLO('yolov13n.yaml')

# 训练模型,参数与论文保持一致
results = model.train(
  data='coco.yaml',
  epochs=600, 
  batch=256, 
  imgsz=640,
  scale=0.5,  # S:0.9; L:0.9; X:0.9
  mosaic=1.0,
  mixup=0.0,  # S:0.05; L:0.15; X:0.2
  copy_paste=0.1,  # S:0.15; L:0.5; X:0.6
  device="0,1,2,3",
)

推理测试

使用训练好的模型进行目标检测:

from ultralytics import YOLO

# 加载模型
model = YOLO('yolov13n.pt')

# 对图像进行推理
results = model("path/to/your/image.jpg")
results[0].show()  # 显示检测结果

实验结果对比

MS COCO数据集性能

论文中报告的YOLOv13与其他实时目标检测器在MS COCO数据集上的性能对比:

方法FLOPs (G)参数 (M)AP50:95valAP50valAP75val延迟 (ms)
YOLO11-N6.52.638.654.241.61.53
YOLOv12-N6.52.640.156.043.41.83
YOLOv13-N6.42.541.657.845.11.97
YOLO11-S21.59.445.862.649.82.56
YOLOv12-S21.49.347.164.251.02.82
YOLOv13-S20.89.048.065.252.02.98

模型导出

将模型导出为ONNX或TensorRT格式,以便在不同平台部署:

from ultralytics import YOLO

model = YOLO('yolov13n.pt')
model.export(format="onnx")  # 导出为ONNX格式
# model.export(format="engine", half=True)  # 导出为TensorRT格式

总结

本指南介绍了Yolov13论文实验数据的获取方法和代码复现步骤。通过使用提供的预训练模型权重和遵循训练配置,可以复现论文中报告的性能结果。项目的技术文档可参考README.md,其中详细描述了模型架构和实验设置。如有问题,欢迎通过项目渠道反馈。

【免费下载链接】Yolov13 【免费下载链接】Yolov13 项目地址: https://ai.gitcode.com/hf_mirrors/atalaydenknalbant/Yolov13

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值