一种基于改进YOLOv5L+DeepSort的船舶机舱火灾实时检测系统及方法

摘要

本发明提供一种基于改进YOLOv5L+DeepSort的船舶机舱火灾实时检测方法。本发明包括：获取机舱火灾图像和视频数据；把采集到的火灾图像输入到改进的YOLOv5L模型进行特征提取，获取特征图，并将输入图像分割成若干大小相同的栅格；对疑似火灾区域给出多个预测框；利用非极大值抑制方法将置信度区间不符合要求的预测框剔除掉，得到最优目标框，获得目标位置；将YOLOv5L检测到的目标位置信息和目标框信息同步到DeepSort目标跟踪算法；DeepSort目标跟踪算法通过卡尔曼滤波器来预测火焰目标运动趋势从而得到火焰蔓延趋势。本发明改进YOLOv5L模型与DeepSort多目标跟踪算法进行融合，将YOLOv5L检测到火灾位置信息传递给DeepSort，对火焰目标进行实时跟踪并预测其未来蔓延趋势，及时的指导机舱工作人员进行定向灭火。