许可协议: agpl-3.0 标签:

• 目标检测
• 计算机视觉
• yolov10
• pytorch_model_hub_mixin 数据集:
• detection-datasets/coco 库名称: yolov10 推理: false

模型描述

YOLOv10：实时端到端目标检测

• 预印本: https://arxiv.org/abs/2405.14458v1
• GitHub: https://github.com/THU-MIG/yolov10

安装指南

pip install git+https://github.com/THU-MIG/yolov10.git

训练与验证

from ultralytics import YOLOv10

model = YOLOv10.from_pretrained('jameslahm/yolov10m')
# 训练
model.train(...)
# 训练完成后可推送至模型中心
model.push_to_hub("your-hf-username/yolov10-finetuned")

# 验证
model.val(...)

推理示例

以下是在猫咪图片上进行端到端推理的完整示例：

from ultralytics import YOLOv10

model = YOLOv10.from_pretrained('jameslahm/yolov10m')
source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
model.predict(source=source, save=True)

效果展示：

BibTeX引用信息

@article{wang2024yolov10,
 title={YOLOv10: Real-Time End-to-End Object Detection},
 author={Wang, Ao and Chen, Hui and Liu, Lihao and Chen, Kai and Lin, Zijia and Han, Jungong and Ding, Guiguang},
 journal={arXiv preprint arXiv:2405.14458},
 year={2024}
}