基于Swin Transformer和PJF颜色空间的火灾检测算法

打开文本图片集

摘  要： 为了更加准确和可靠地在监控图像中对火灾进行早期监测，以便辅助火焰检测仪器更加高效和准确地检测火焰，文中基于Transformer算法和PJF颜色空间，设计了一种深度学习的火灾图像检测算法。首先，对数据集中的火灾图像数据进行预处理，将图像从RGB颜色空间转化为PJF颜色空间，突出显示图像的火焰区域；然后，将预处理后的图像输入到Swin Transformer网络中，过程中采用注意力机制、移动窗口机制、相对位置偏移机制对火灾图像进行特征提取；最后，将提取的图像特征送入全连接层，获取火灾图像的检测结果。（剩余9509字）