ARTICLE DETAIL

资讯详情

深耕网站建设与运营推广的一线实战洞察。

YOLO牧场围栏内小牛与成年牛目标检测数据集-12106张

YOLO牧场围栏内小牛与成年牛目标检测数据集-12106张

YOLO牧场围栏内小牛与成年牛目标检测数据集

小牛智能识别:YOLO26 训练全流程

📊 数据集基本信息

  • 目标类别: [‘0’, ‘1’]
  • 中文类别:[‘小牛’, ‘成年牛’]
  • 训练集:10587 张
  • 验证集:1013 张
  • 测试集:506 张
  • 总计:12106 张

📄 data.yaml 配置信息

该数据集提供了data.yaml文件,内容如下:

train:../train/imagesval:../valid/imagestest:../test/imagesnc:2names:['0','1']

🖼️ 标注可视化



📦 数据集下载

🚀👉 立即下载完整数据集👈

✅ 包含全部train / val / test图像与 YOLO 标签,可直接训练。

数据集详细介绍

该数据集专为牧场环境下的目标检测任务设计,名称明确指向其应用场景——即对牧场围栏内的小牛与成年牛进行识别与定位。从类别名称可推断,该数据集主要用于农业自动化、牲畜管理及智能监控系统中,帮助实现对牛群的实时监测与分类管理。在实际应用中,此数据集可用于开发自动化的放牧管理系统、牲畜健康追踪平台或智能安防监控系统,提升牧场管理的效率和精准度。

数据集包含共计12106张图像,覆盖两种主要类别,分别对应小牛和成年牛。虽然未提供具体的英文类别名称,但根据中文描述可推测其分别为“calf”和“成年牛”或类似表述。数据集中两类对象的分布可能具有一定的不平衡性,具体比例需进一步分析,但整体上涵盖了牧场环境中常见的牛类个体,具备良好的代表性。图像采集场景可能包括不同光照条件、天气状况以及牛只姿态变化,增强了数据集的多样性和实用性。

标注规范方面,该数据集采用标准的目标检测标注格式,每张图像均配有边界框标注,确保模型能够准确学习到目标物体的位置信息。标注过程可能遵循统一的尺度标准和类别定义,保证了数据的一致性和可靠性。这种高质量的标注为训练高精度的目标检测模型提供了坚实的基础,尤其适用于YOLO等轻量级检测框架的优化与部署。

在典型应用方向上,该数据集可广泛用于畜牧领域的智能化升级,如自动计数、行为分析、异常检测等。此外,结合边缘计算设备,可实现低延迟的实时检测,适用于牧场现场部署。对于研究者而言,建议在实际应用中结合多模态数据(如红外图像或音频信息)以提升检测鲁棒性,并探索迁移学习方法以适应不同牧场环境的变化。

YOLO26 目标检测算法原理

评价一个检测模型不能只看 mAP 一个数字。YOLO26 的跨版本对比表里暗藏玄机——它用略低的参数量跑出了不输大模型的精度,靠的就是架构简化带来的效率红利。

端到端无 NMS 推理:传统 YOLO 模型在推理后需要执行非极大值抑制(NMS)来去除冗余检测框,这增加了后处理延迟和部署复杂度。YOLO26 默认采用一对一(one-to-one)检测头,直接输出每张图像最多 300 个检测结果(N, 300, 6),完全省去 NMS 步骤,推理流水线大幅简化。在 CPU 上的 ONNX 推理速度相比 YOLO11n 提升高达 43%。

无 DFL 回归:YOLO26 移除了分布焦点损失(Distribution Focal Loss, DFL)结构,检测头更加轻量。边界框回归不再受限于预设的离散区间,既降低了模型复杂度又保持了同等甚至更优的回归精度。

Progressive Loss 与 STAL:训练过程中通过渐进损失(Progressive Loss)将监督信号逐步从辅助头部转移到推理阶段的主头部。小目标感知标签分配(STAL)策略特别提升小目标的正样本标签覆盖率,改善小物体检测性能。

MuSGD 混合优化器:创新性地将大语言模型常用的 Muon 优化器与 SGD 结合,实现更稳定高效的训练收敛。相比纯 SGD,MuSGD 在大 batch 训练时收敛更快且不易震荡。

模型尺度与 COCO 性能:YOLO26 提供 N/S/M/L/X 五种尺度:

模型mAP50-95CPU ONNX(ms)T4 TensorRT(ms)参数量
YOLO26n40.938.91.72.4M
YOLO26s48.687.22.59.5M
YOLO26m53.1220.04.720.4M
YOLO26l55.0286.26.224.8M
YOLO26x57.5525.811.855.7M

Nano 版本参数量仅 2.4M,CPU 推理速度比 YOLO11n 快 43%,适合边缘设备和小数据集;X 版本 mAP 达 57.5,适用于对准确率要求极高的场景。

双头架构设计:YOLO26 检测模型内置两个检测头——一对一(默认,无 NMS,输出 300 个检测框)和一对多(需 NMS,输出 8400 个候选框)。训练时一对多头作为辅助监督信号加速收敛,推理时默认使用一对一头实现端到端输出。

YOLO26 支持全部七大视觉任务:检测、实例分割、语义分割、深度估计、分类、姿态估计和定向边界框检测(OBB),一个框架覆盖从 2D 检测到 3D 感知的完整需求。

关键训练参数配置详解

参数调优不是玄学——batch 看显存、lr0 看 batch、epochs 看数据量、imgsz 看目标大小。记住这四个对应关系就够用了。

参数默认值本项目设置含义
model-yolo26n.pt预训练权重,n/s/m/l/x 五种尺度
epochs100150训练总轮数
imgsz640416输入图像尺寸
batch1664批次大小,根据显存调整
lr00.010.02初始学习率
lrf0.010.01最终学习率因子(lr0 × lrf)
momentum0.9370.937SGD 动量
weight_decay0.00050.0005权重衰减(L2 正则化)
warmup_epochs3.03.0预热轮数
warmup_momentum0.80.8预热期间动量初始值
box7.57.5边界框回归损失权重
cls0.50.5分类损失权重
dfl1.51.5DFL 损失权重(YOLO26 可置 0)
patience100-1早停轮数
cos_lrFalseTrue余弦学习率衰减
close_mosaic1010最后 N 轮关闭 Mosaic 增强
optimizerautoSGD优化器选择
ampTrueTrue自动混合精度训练
freezeNone0冻结前 N 层骨干网络参数
dropout0.00.1分类头 Dropout 率,防过拟合

训练日志里的 box_loss/cls_loss 比值可以告诉你模型在偏重什么。比值 > 10 说明定位损失占主导,比值 < 3 说明分类问题更大。

学习率调参经验

训练可视化与TensorBoard:YOLO26 默认开启 TensorBoard 日志记录,训练过程中可实时查看 loss 曲线和指标变化。在命令行执行tensorboard --logdir runs/detect/train即可打开 Web 看板,每轮训练完成后自动刷新。训练结束后 results.png 生成的六合一曲线图是最直观的诊断工具——一眼就能看出是否过拟合、是否收敛、哪个 loss 在震荡。

断点续训机制:如果训练意外中断(断电、OOM 等),不用从头开始。YOLO 每轮保存last.pt,恢复训练时指定model=last.pt并设置resume=True,学习率和优化器状态都会从断点恢复,训练曲线也和连续训练一样平滑。

模型评估指标解读

评估一个检测模型,光看 mAP50 远远不够。mAP50-95、Precision、Recall 这三个数放一起看,才能拼出完整画面。

训练完成后,YOLO26 训练器自动输出以下指标:

mAP50(mAP@0.5):IoU 阈值为 0.5 时的平均精度均值,反映模型粗粒度定位能力。mAP50 ≥ 0.90 表示定位效果很好。

mAP50-95(mAP@0.5:0.95):IoU 从 0.5 到 0.95 共 10 个阈值上取平均。更严苛,反映精确定位能力,是更具代表性的综合评价指标。

Precision(精确率):检测结果中正确检测的比例。高 Precision 意味着虚警少。

Recall(召回率):真实目标中被成功检测到的比例。高 Recall 意味着漏检少。

F1 Score:Precision 和 Recall 的调和平均数 = 2×P×R/(P+R),综合衡量模型平衡性。

Confusion Matrix(混淆矩阵):N×N 矩阵,对角线为正确分类,非对角线为误分类,可定位类别间混淆情况。

训练曲线分析

指标使用场景

对比不同训练版本的 PR 曲线时,不要只看曲线位置高低——看形状变化更重要。训练后期曲线从陡峭变平滑是好事,说明模型鲁棒性在提升。

YOLO 系列算法演进史

如果只从架构复杂度看,YOLO26 的检测头比 YOLOv8 还要简单——参数更少、后处理更少,但精度不降。这种「减法式进化」在 YOLO 历史上是第一次。

YOLOv1 (2015):将检测问题建模为单一回归问题,直接在输出层预测边界框和类别概率。速度极快但定位精度较差。

YOLOv2/YOLOv3 (2016-2018):引入 anchor box 机制、特征金字塔(FPN)和多尺度训练。YOLOv3 的 Darknet-53 骨干网络成为经典架构,至今仍被广泛使用。

YOLOv5 (2020):Ultralytics 在 PyTorch 上的实现,带来了完善的训练框架、自动锚框聚类和数据增强 pipeline,大幅降低了使用门槛。

YOLOv8 (2023):统一了检测、分割、分类和关键点任务的框架,引入 C2f 模块和无锚框(anchor-free)检测头。

YOLOv9 (2024):提出 GELAN(通用高效层聚合网络)和 PGI(可编程梯度信息),同等参数量下精度提升显著。

YOLOv10/YOLOv11 (2024):YOLOv10 首次提出无 NMS 端到端检测。YOLOv11 进一步优化骨干网络和训练策略,在速度和精度之间取得新平衡。

YOLO26 (2026):集历代之大成,核心创新包括:① 端到端无 NMS 一对一检测头 ② 无 DFL 的简化回归头 ③ Progressive Loss 渐进式监督转移 ④ STAL 小目标感知标签分配 ⑤ MuSGD 混合优化器。

关键设计演进总结

YOLO26 训练步骤详解

环境搭好后,训练就是在 data.yaml 和模型之间建立一条通路。下面是完整的操作步骤。

环境准备

pipinstallultralytics yolo checks python-c"import torch; print(torch.cuda.is_available())"

命令行训练

yolo detect traindata=data.yamlmodel=yolo26n.ptepochs=150imgsz=416batch=64device=0lr0=0.02patience=-1cos_lr=Trueclose_mosaic=10

Python API 训练

fromultralyticsimportYOLO model=YOLO("yolo26n.pt")results=model.train(data="data.yaml",epochs=150,imgsz=416,batch=64,device=0,lr0=0.02,patience=-1,cos_lr=True,close_mosaic=10,augment=True,amp=True,workers=8,)

训练输出文件

单张推理测试

model=YOLO("runs/detect/train/weights/best.pt")results=model("test_image.jpg")results[0].show()

多 GPU 分布式训练

yolo detect traindata=data.yamlmodel=yolo26n.ptdevice=0,1epochs=100

恢复中断的训练

yolo detect trainmodel=path/to/last.ptdata=data.yamlresume=True

部署与应用建议

生产环境部署:大数据集训练出的模型泛化能力较强,可直接导出多种格式用于不同场景:

fromultralyticsimportYOLO model=YOLO("runs/detect/train/weights/best.pt")# GPU 服务器:TensorRT(最快)model.export(format="engine",device=0,imgsz=416)# 跨平台:ONNXmodel.export(format="onnx",imgsz=416)# 移动端:TFLite int8 量化model.export(format="tflite",imgsz=416,int8=True)

持续优化:可定期收集线上误检/漏检样本纳入训练集,迭代提升模型表现。大数据集下知识蒸馏或多尺度训练也能带来额外精度收益。

通用训练经验:12106 张图训练目标检测模型,遇到的问题是典型的——数据不多不少,精度不上不下。

这种规模的数据集,调参的边际收益其实很高。几个值得优先尝试的方向:

一是学习率的余弦退火。默认 YOLO26 用的是 cos_lr 衰减,但周期很关键。如果训练 80 轮,lr0=0.01搭配cos_lr=True,学习率会从 0.01 平滑降到接近 0。这个曲线比阶梯式衰减更适合中等数据集,收敛更稳定。

二是mosaic=1.0时最后 10 轮关闭(close_mosaic=10)。这个设置让模型在训练末期接触真实数据分布,对部署时的精度很重要。很多人忽视这个参数,训完发现部署精度掉了一大截。

三是关注训练过程中的val_box_loss拐点。通常 30-40 轮附近会出现一个平台期,如果 50 轮后仍无明显下降,基本可以判断当前参数组合不够理想。

你可能还需要知道的几个细节:YOLO26 的文档很全,但有些实战经验不在文档里。

imgsz=416imgsz=640的差异比你想象的大。416 下每张图的训练时间短大约 30%,但对小目标的召回率可能差 5 个点以上。如果你的数据集中小目标(小于 32×32 像素)占比超过 30%,建议用 512 或 640。

② AMP(自动混合精度)默认开启,但这在 GTX 16 系列等不支持 TensorCore 的 GPU 上反而拖慢训练。如果遇到训练速度异常慢,试试关掉 AMP:amp=False

close_mosaic=10意味着最后 10 轮关闭 Mosaic 数据增强。这个设置很关键——如果不关,模型始终在"拼贴图"上评估,没见过真实数据的分布,最终部署时精度会掉一大截。

④ 训练数据里有大量重复或高度相似的图片时(比如视频截帧),split='val'自动划分可能导致训练和验证集高度相似,验证指标虚高。建议手动划分或使用按场景/时间分割的策略。

workers=8是默认值,但如果你的硬盘是机械硬盘而非 SSD,并发读图可能会成为瓶颈。CPU 占用率满但 GPU 利用率低时,试着降低 workers 到 2-4。

返回列表