基于视觉变换网络的自监督森林火灾识别 |
| |
引用本文: | 宁重阳,汤梓桐,谢亦秋,戴颖成,李建军.基于视觉变换网络的自监督森林火灾识别[J].中南林业科技大学学报,2023(7):32-40. |
| |
作者姓名: | 宁重阳 汤梓桐 谢亦秋 戴颖成 李建军 |
| |
作者单位: | 1. 中南林业科技大学计算机与信息工程学院;2. 中国林业科学研究院资源信息研究所 |
| |
基金项目: | 国家自然科学基金项目(31570627); |
| |
摘 要: | 【目的】森林火灾识别是避免森林火灾大面积蔓延的一项重要研究。随着深度学习的快速发展,基于卷积神经网络的模型因其在图像识别领域的优异表现,被广泛应用到森林火灾识别任务当中。然而,基于卷积神经网络的方法通常在标签数据不充分时,难以取得令人满意的森林火灾识别结果。【方法】本研究提出了一种基于视觉变换网络的自监督森林火灾识别模型(Self supervised forest fire identification model based on visual transformation network),来提高模型在标签稀缺情况下的森林火灾识别精度。具体来说,该模型采用视觉变换网络作为主干网络,通过视觉变换网络中的多头自注意力机制来捕获森林火灾图像的全局信息特征。并且引入自监督学习中的图像重建任务来辅助模型训练,从而减少模型对标签数据的依赖。模型通过对掩盖图像的特征恢复和重建学习相关语义信息。同时,本研究还提出了一种基于傅里叶低频混合变换的数据增强方法来提高模型的泛化性和鲁棒性。【结果】通过开展详细的试验来验证模型的有效性,结果表明,与其他常见的网络模型相比,FFDM模型在森林火灾识别任务中取...
|
关 键 词: | 森林火灾识别 傅里叶变换 自监督学习 视觉变换网络 低频变换 |
|
|