一种基于卷积神经网络的视频野外火灾烟雾实时检测方法

摘要

本发明提出了一种基于卷积神经网络的视频野外火灾烟雾实时检测方法。本发明通过实验模拟方式收集烟雾图像数据集并创建训练集、测试集及验证集；分别对训练集、测试集及验证集进行自动标注结合人工调整，分别得到带真实标签的训练集、测试集以及验证集；对带真实标签的训练集以及验证集进行图像旋转处理、颜色通道加减色处理、缩放处理得到处理后带真实标签的训练集、验证集；初始化卷积神经网络参数，根据缩放处理后带真实标签的训练集训练建立好的卷积神经网络模型；实时采集待检测野外监控画面图像，通过训练后卷积神经网络模型预测烟雾目标检测框并进行优化；在训练后卷积神经网络模型给出的目标检测结果上进行帧间置信度增强以及重定位。