一种基于注意力机制的早期烟雾检测方法

本发明公开了一种基于注意力机制的早期烟雾检测方法，收集来自不同场景下的烟雾图像，通过手动或自动标记的方式，构建烟雾图像数据集，按比例划分训练集和测试集；构建基于注意力模块和不同复杂度的卷积模块组成的网络结构，用于提取不同层次的特征信息。其次，引入自适应空间特征融合模块，动态调整不同尺寸特征的权重，通过卷积操作计算各尺寸特征的权重，生成特征图。构建检测头，负责预测目标的边界框、置信度以及类别。检测头会根据特征图中的信息生成最终的检测结果。本发明能够使得早期烟雾检测的准确率提高，具有很好的检测效果。