基于CN和三通道胶囊网络的森林火灾识别方法及装置
摘要:
本发明公开了一种基于CN和三通道胶囊网络的森林火灾识别方法和装置,识别方法包括以下步骤:构建森林火灾火焰的初始样本集合;获取初始样本集合中的每个样本的火焰区域图像,构建形成火焰样本集合;创建三通道CapsNet网络模型;通过Mnist数据集对单个的CapsNet网络进行训练,将对单个的CapsNet网络进行训练形成的参数迁移至三通道CapsNet网络模型中,三通道CapsNet网络模型中的总全连接层的参数通过随机初始化实现;通过火焰样本集合对三通道CapsNet网络模型进行二次训练,形成最终的火灾识别模型;应用CN算法构建火焰样本集合描述的主元颜色空间向量。本发明提高了火灾检测的实时性和有效性。
0/0