买专利卖专利找龙图腾，真高效！ 查专利查商标用IPTOP,全免费！专利年费监控用IP管家,真方便！

申请/专利权人：北京信息科技大学

摘要：本发明公开了一种基于改进的YOLOv5模型的火焰检测方法，包括：建立火焰数据集；以YOLOv5模型为基础模型，嵌入注意力机制单元、增加新增检测层、引入边界损失函数得到第一YOLOv5模型；对第一YOLOv5模型进行预训练得到第二YOLOv5模型；对第二YOLOv5模型进行优化，得到第三YOLOv5模型；将测试图像输入第三YOLOv5模型得到火焰检测结果。通过嵌入注意力机制单元，能够同时考虑通道间关系和空间信息，使得第一YOLOv5模型能够更准确地定位并识别目标区域；增加新增检测层用于检测小尺寸目标；引入边界损失函数对小数据集和噪声的鲁棒性更强，并且更适合轻量化模型。

主权项：1.一种基于改进的YOLOv5模型的火焰检测方法，其特征在于，包括：建立火焰数据集，所述火焰数据集包括多张火焰图像；以YOLOv5模型为基础模型，所述YOLOv5模型包括顺次设置的输入端、基准网络模块、特征融合模块和检测模块，在所述基准网络模块中嵌入注意力机制单元、在所述检测模块增加新增检测层、在所述检测模块引入边界损失函数得到第一YOLOv5模型，所述边界损失函数按照以下方法进行计算： 其中，为所述火焰图像的预测边框与结果边框的差异；IOU为交互比，α为调节参数；所述基准网络模块包括第一单元、第二单元、第三单元和第四单元；所述第一单元为切片结构；所述第二单元由卷积层、批量归一化和第一激活函数组成；所述第三单元由所述卷积层和瓶颈层组成；所述第四单元由所述卷积层与池化层组成；所述基准网络模块提取测试图像的特征信息得到第一特征图；所述在所述基准网络模块中嵌入注意力机制单元，包括：所述注意力机制单元通过沿第一方向和第二方向分别对所述第一特征图进行平均池化得到第一向量和第二向量，对所述第一向量和所述第二向量进行通道融合后利用卷积块进行通道压缩、编码所述第一特征图沿所述第一方向和所述第二方向的空间信息后分离通道、通过所述卷积块得到与所述第一特征图通道数相同的第二特征图，通过第二激活函数进行归一化加权；所述在所述检测模块增加新增检测层，包括：所述新增检测层的尺寸为160×160，用于检测感受野尺寸大于4×4的目标；所述特征融合模块包括24层特征提取层，增加2层所述特征融合模块中的所述特征提取层匹配所述检测层，所述第二特征图经过所述特征融合模块的第二层所述特征提取层处理得到第三特征图，所述第二特征图经过所述特征融合模块的第二十六层所述特征提取层处理得到第四特征图，所述第三特征图与所述第四特征图融合得到第五特征图；所述新增检测层对所述第五特征图进行检测；采用图像数据集对所述第一YOLOv5模型进行预训练，得到第二YOLOv5模型；将所述火焰数据集划分为训练集、验证集和测试集，所述测试集中的所述火焰图像为测试图像，将所述训练集和所述验证集输入所述第二YOLOv5模型进行优化，得到第三YOLOv5模型；将所述测试集的所述测试图像输入所述第三YOLOv5模型，得到火焰检测结果。

全文数据：

权利要求：

百度查询： 北京信息科技大学 基于改进的YOLOv5模型的火焰检测方法