YOLO海上漂浮物浮标与垃圾目标检测数据集
大规模浮标检测:YOLO26 训练与评估
📊 数据集基本信息
- 目标类别: [‘buoy’, ‘trash’]
- 中文类别:[‘浮标’, ‘垃圾’]
- 训练集:2097 张
- 验证集:200 张
- 测试集:100 张
- 总计:2397 张
📄 data.yaml 配置信息
该数据集提供了data.yaml文件,内容如下:
train:../train/imagesval:../valid/imagestest:../test/imagesnc:2names:['buoy','trash']🖼️ 标注可视化
数据集下载
数据集下载:⬇️⬇️⬇️ 点击下载
数据集详细介绍
YOLO海上漂浮物浮标与垃圾目标检测数据集是一个面向海洋环境监测和自动化识别的专用数据集,其名称直接反映了其应用场景。该数据集主要用于检测海面上的漂浮物,包括浮标和垃圾两类目标,适用于海洋污染监控、船舶自动避障、海洋生态保护等实际场景。在近海或远洋作业中,及时识别并处理漂浮物对于保障航行安全和生态环境具有重要意义,因此该数据集可作为训练和验证相关检测模型的重要资源。
数据集包含2397张图像,覆盖多种光照条件和天气状况下的海面场景,确保了模型的泛化能力。类别数量为2类,分别对应“浮标”和“垃圾”,其中“浮标”通常指用于导航或监测的设备,“垃圾”则涵盖塑料瓶、包装袋、废弃渔网等常见海洋污染物。这种分类方式使得模型能够区分不同类型的漂浮物,从而支持后续的分类与处理决策。
标注规范方面,数据集采用标准的边界框标注方式,每个目标均被精确标定,标注过程遵循统一的尺度和格式要求,保证了数据的一致性和可用性。这种高质量的标注为模型训练提供了可靠的基础,有助于提升检测精度和鲁棒性,尤其在复杂背景下的小目标检测中表现更为突出。
该数据集适合用于开发基于YOLO系列模型的实时目标检测系统,尤其适用于无人机、无人船或固定摄像头等平台的部署。在实际应用中,建议结合多源数据进行融合分析,例如与水文数据、气象信息结合,以提高检测的准确性和实用性。此外,还可通过数据增强和迁移学习策略进一步优化模型性能,拓展其在不同海域环境中的适用性。
YOLO26 目标检测算法原理
2026 年 Ultralytics 放出 YOLO26 的时候,社区最关注的不是精度涨了几个点,而是"端到端无 NMS"这个架构变革。
端到端无 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-95 | CPU ONNX(ms) | T4 TensorRT(ms) | 参数量 |
|---|---|---|---|---|
| YOLO26n | 40.9 | 38.9 | 1.7 | 2.4M |
| YOLO26s | 48.6 | 87.2 | 2.5 | 9.5M |
| YOLO26m | 53.1 | 220.0 | 4.7 | 20.4M |
| YOLO26l | 55.0 | 286.2 | 6.2 | 24.8M |
| YOLO26x | 57.5 | 525.8 | 11.8 | 55.7M |
Nano 版本参数量仅 2.4M,CPU 推理速度比 YOLO11n 快 43%,适合边缘设备和小数据集;X 版本 mAP 达 57.5,适用于对准确率要求极高的场景。
双头架构设计:YOLO26 检测模型内置两个检测头——一对一(默认,无 NMS,输出 300 个检测框)和一对多(需 NMS,输出 8400 个候选框)。训练时一对多头作为辅助监督信号加速收敛,推理时默认使用一对一头实现端到端输出。
YOLO26 支持全部七大视觉任务:检测、实例分割、语义分割、深度估计、分类、姿态估计和定向边界框检测(OBB),一个框架覆盖从 2D 检测到 3D 感知的完整需求。
关键训练参数配置详解
训练参数调对了一件事半功倍,调错了折腾几天也出不了好结果。这里把每个关键参数的含义、默认值和推荐设置都列清楚了。
| 参数 | 默认值 | 本项目设置 | 含义 |
|---|---|---|---|
model | - | yolo26n.pt | 预训练权重,n/s/m/l/x 五种尺度 |
epochs | 100 | 150 | 训练总轮数 |
imgsz | 640 | 416 | 输入图像尺寸 |
batch | 16 | 64 | 批次大小,根据显存调整 |
lr0 | 0.01 | 0.02 | 初始学习率 |
lrf | 0.01 | 0.01 | 最终学习率因子(lr0 × lrf) |
momentum | 0.937 | 0.937 | SGD 动量 |
weight_decay | 0.0005 | 0.0005 | 权重衰减(L2 正则化) |
warmup_epochs | 3.0 | 3.0 | 预热轮数 |
warmup_momentum | 0.8 | 0.8 | 预热期间动量初始值 |
box | 7.5 | 7.5 | 边界框回归损失权重 |
cls | 0.5 | 0.5 | 分类损失权重 |
dfl | 1.5 | 1.5 | DFL 损失权重(YOLO26 可置 0) |
patience | 100 | -1 | 早停轮数 |
cos_lr | False | True | 余弦学习率衰减 |
close_mosaic | 10 | 10 | 最后 N 轮关闭 Mosaic 增强 |
optimizer | auto | SGD | 优化器选择 |
amp | True | True | 自动混合精度训练 |
freeze | None | 0 | 冻结前 N 层骨干网络参数 |
dropout | 0.0 | 0.1 | 分类头 Dropout 率,防过拟合 |
YOLO26 训练时 loss 曲线的下降速度和最终值受 lr0 和 weight_decay 影响最大。如果 loss 从一开始就降得很慢,大概率是 lr0 设低了。
学习率调参经验:
- 训练初期 loss 不下降 → 调大
lr0或增加warmup_epochs - 训练后期 loss 震荡 → 降低
lr0,增大weight_decay - 验证集 loss 先降后升(过拟合)→ 增大
dropout、weight_decay或减小epochs - 小数据集(<100张)→
lr0=0.001,weight_decay=5e-4,dropout=0.2 - 大数据集(>1000张)→
lr0=0.01,weight_decay=5e-4,batch=32+
训练可视化与TensorBoard:YOLO26 默认开启 TensorBoard 日志记录,训练过程中可实时查看 loss 曲线和指标变化。在命令行执行tensorboard --logdir runs/detect/train即可打开 Web 看板,每轮训练完成后自动刷新。训练结束后 results.png 生成的六合一曲线图是最直观的诊断工具——一眼就能看出是否过拟合、是否收敛、哪个 loss 在震荡。
断点续训机制:如果训练意外中断(断电、OOM 等),不用从头开始。YOLO 每轮保存last.pt,恢复训练时指定model=last.pt并设置resume=True,学习率和优化器状态都会从断点恢复,训练曲线也和连续训练一样平滑。
模型评估指标解读
模型训完了,接下来怎么判断它好不好用?YOLO 的输出文件夹里有一整套评估文件,逐个看懂才算真正了解模型水平。
训练完成后,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 矩阵,对角线为正确分类,非对角线为误分类,可定位类别间混淆情况。
训练曲线分析:
train/box_loss单调下降 → 边界框回归正常收敛val/box_loss先降后升 → 过拟合风险,应增大数据增强或减小模型metrics/mAP50(B)增长放缓趋于平稳 → 模型接近收敛上限- train 持续降、val 持续升 → 严重过拟合,建议减小模型或增大 dropout
指标使用场景:
- 安全生产场景(如安全帽检测)→ 优先保证 Recall,宁可多报不能漏报
- 质检场景(如缺陷检测)→ 优先保证 Precision,降低误检率减少人工复核量
- 通用检测 → 均衡优化,关注 mAP50-95 和 F1 Score
如果 PR 曲线在中段(Recall 0.5-0.7)出现明显凹陷,通常意味着模型对某些难例的置信度摇摆不定。这种情况加 hard negative mining 或调高 conf 阈值缓解不了,需要扩充难例样本。
YOLO 系列算法演进史
了解 YOLO 的发展脉络有助于理解为什么 YOLO26 的架构是这样设计的。每个大版本都不是孤立出现的,而是对上一个版本痛点的回应。
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 混合优化器。
关键设计演进总结:
- Anchor-based → Anchor-free(v8 起)
- C3 → C2f → C3k2(骨干模块持续优化)
- 多任务统一框架(v8 起)
- NMS 后处理 → 无 NMS 端到端(v10/v26)
- 单一损失函数 → 多任务联合损失(Box + Cls + DFL,v26 可去 DFL)
YOLO26 训练步骤详解
从环境搭建到模型推理,整个训练链路大概需要这些操作
环境准备:
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=10Python 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,)训练输出文件:
runs/detect/train/weights/best.pt:验证集上 mAP 最高的权重runs/detect/train/weights/last.pt:最后一轮保存的权重runs/detect/train/results.csv:每轮 loss 和指标数据runs/detect/train/confusion_matrix.png:混淆矩阵图runs/detect/train/results.png:训练曲线图runs/detect/train/val_batch*_pred.jpg:验证集预测效果图
单张推理测试:
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部署与应用建议
有了 2397 张图打底,模型的底子已经比较扎实。部署时重点关注两个指标:推理延迟(P99 < 100ms)和内存占用(< 500MB)。
model=YOLO("runs/detect/train/weights/best.pt")# NVIDIA GPU: TensorRT + FP16model.export(format="engine",device=0,half=True,workspace=4)# 跨平台/纯 CPU: ONNXmodel.export(format="onnx",imgsz=416)# Intel CPU: OpenVINO(比 ONNX Runtime 快 15-30%)model.export(format="openvino",imgsz=416)大规模部署架构:如果需要在多台服务器上部署,建议搭配 Triton Inference Server 做模型管理和负载均衡,支持动态 batching 和模型版本热切换。YOLO26 的 ONNX/TensorRT 导出版本都可以直接接入 Triton 的模型仓库。
通用训练经验:2397 张图训练目标检测模型,遇到的问题是典型的——数据不多不少,精度不上不下。
这种规模的数据集,调参的边际收益其实很高。几个值得优先尝试的方向:
一是学习率的余弦退火。默认 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=416和imgsz=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。