1. 项目背景与核心价值
在公共场所禁烟监管领域,传统人工巡查方式存在效率低、覆盖范围有限等痛点。我们开发的这套系统采用YOLOv11目标检测算法,实现了对吸烟行为的自动化识别与预警。相比市面常见方案,本系统具有三大突破点:
- 采用最新发布的YOLOv11模型,在自建数据集上达到98.7%的识别准确率
- 开发了完整的用户交互系统,支持多级权限管理
- 提供从模型训练到部署的全套解决方案
实际测试表明,在办公室、餐厅等复杂场景下,系统对香烟、电子烟的识别率比传统方案提升40%以上
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 技术架构解析
2.1 YOLOv11模型优化
我们在原生YOLOv11基础上进行了三项关键改进:
-
注意力机制增强:
- 在Backbone末端添加CBAM模块
- 针对小目标检测优化特征融合方式
python复制class EnhancedCBAM(nn.Module): def __init__(self, channels): super().__init__() self.ca = ChannelAttention(channels) self.sa = SpatialAttention() def forward(self, x): x = self.ca(x) * x x = self.sa(x) * x return x -
数据增强策略:
- 采用Mosaic+MixUp组合增强
- 针对烟雾形态设计弹性变换
-
损失函数改进:
- 使用WIoU替代CIoU
- 分类分支引入Focal Loss
2.2 数据集构建
我们收集了超过5万张标注图像,涵盖:
- 不同光照条件(白天/夜晚/逆光)
- 各类吸烟姿态(手持/叼烟/电子烟)
- 多种干扰场景(食物/笔/手机等相似物)
数据集采用COCO格式标注,包含以下关键类别:
| 类别 | 样本量 | 难点说明 |
|---|---|---|
| 香烟 | 28,742 | 存在部分遮挡 |
