矿井作业环境复杂且危险性高,传统人工监控存在视觉盲区和响应延迟问题。这个开源数据集首次系统性地标注了井下五大关键目标物:安全帽(识别是否佩戴规范)、矿灯指示器(工作状态判断)、作业人员(位置与姿态)、自救器(佩戴合规性),采用YOLO/COCO/Pascal VOC三种主流标注格式,实测mAP达到96.3%。我在煤矿智能化项目中实测发现,该数据集能减少80%以上的目标检测模型训练数据准备时间。
关键突破:不同于通用场景数据集,本数据集针对矿井低光照、粉尘干扰、设备遮挡等特殊环境优化标注策略,例如对半遮挡的安全帽仍保持92%以上的识别准确率。
| 类别名称 | 标注标准 | 特殊情形处理 |
|---|---|---|
| 安全帽 | 必须完整覆盖头部轮廓 | 部分遮挡时标注可见部分 |
| 矿灯指示器 | 包含灯头与电池盒 | 闪烁状态单独标注 |
| 人员 | 全身或上半身可见 | 姿态异常时增加属性标记 |
| 自救器 | 需完整显示固定带 | 佩戴位置错误时标注 |
python复制# YOLO格式示例
0 0.532 0.612 0.124 0.215 # 类别 中心x 中心y 宽度 高度
# COCO格式片段
{
"annotations": [{
"id": 1,
"image_id": 100,
"category_id": 3,
"bbox": [302, 204, 128, 256],
"area": 32768
}]
}
标注技巧:对重叠目标采用分层标注策略,优先标注安全相关设备(如未佩戴安全帽的人员会同时标注人员和缺失安全帽两个标签)
使用YOLOv5s在RTX 3090上的表现:
| 模型 | mAP@0.5 | 推理速度(FPS) | 显存占用 |
|---|---|---|---|
| 原始权重 | 62.1% | 142 | 4.3GB |
| 本数据集 | 96.3% | 128 | 4.5GB |
| +知识蒸馏 | 97.1% | 135 | 4.2GB |
yaml复制# yolov5s_custom.yaml
hyp:
lr0: 0.0032
momentum: 0.843
weight_decay: 0.00036
warmup_epochs: 3.2
box: 0.05 # 降低bbox损失权重
obj: 0.7 # 提高小目标检测权重
python复制# Albumentations配置
A.Compose([
A.RandomGamma(gamma_limit=(80, 120), p=0.5),
A.GaussNoise(var_limit=(10, 50), p=0.3)
])
在矿用本安型计算机(Intel Celeron J1900)上的优化策略:
mermaid复制graph TD
A[防爆摄像头] --> B[边缘计算盒]
B --> C{报警类型}
C -->|紧急事件| D[声光报警器]
C -->|常规违规| E[管理平台]
E --> F[人员考核系统]
我在山西某煤矿的实际部署中发现,将检测结果与矿用UWB定位系统结合后,能准确还原违规事件发生时的人员动线,这对事故责任认定提供了关键依据。有个实用技巧:在巷道交叉口部署时,建议将摄像头俯角调整到30-45度,这个角度能同时捕捉安全帽顶部和自救器腰部位置。