复杂环境下小尺度烟火目标检测研究

作者：温秀兰; 焦良葆; 李子康; 姚波; 唐国寅

来源：南京信息工程大学学报(自然科学版), 2022, 1-9.

• 烟火检测
• 改进YOLOv5
• DIoU＿Loss
• 优化加速

针对复杂环境下起火点目标尺寸较小、起火点特征易与实际场景混淆导致烟火检测效率及准确率低等问题，提出了一种基于改进YOLOv5的小尺度烟火目标检测方法。首先，在原始YOLOv5模型输出的第3个检测层上增加第4个检测层，以此获取更大的特征图对小目标进行检测，加强网络模型的特征提取能力。其次，为解决目标在被遮挡的场景中容易出现漏检的问题，将原网络中用于计算目标框损失函数的GIoU＿Loss替换成DIoU＿Loss。最后，利用TensorRT对模型进行压缩和加速优化，并将其部署到Jetson TX2开发板上进行加速推理实验，通过复制增强方法扩充实际烟火场景数据，大量实验结果表明，所提方法用于复杂环境下的小尺度烟火目标检测不仅检测速度快而且精度高，适于推广应用。

• 单位
  南京工程学院; 自动化学院