基于三维卷积网络和自注意力机制的山火检测方法和系统

本申请公开了一种基于三维卷积网络和自注意力机制的山火检测方法和系统，属于图像识别、深度学习等技术领域。该方法包括：构建训练集和测试集；对视频数据进行预处理；构建山火识别模型，包括依次连接的三维卷积网络、基于自注意力机制以及一致性注意力正则化机制的特征提取网络和分类网络；利用训练集和测试集对山火识别模型进行训练和测试，得到山火检测模型；利用山火检测模型进行山火检测。本申请在三维卷积网络的基础上加入了三维注意力机制，能够更有效地提取出时空特征，增强网络的区分能力，同时还利用了一致性注意力正则化，对所得到的时空特征进行特征层面的约束，从而得到具有较强自区分共性的特征，提高了网络识别的准确率。