基于视频的夜间航标灯多目标检测模型

    Video-based multi-target detection model for nighttime buoy lights

    • 摘要: 为了实现夜间航标灯的智能检测,提出一种基于视频的航标灯质检测网络模型,首先利用检测模块捕捉航标灯质的颜色特征,接着利用跟踪模块对灯质颜色特征进行跟踪以锁定目标,再利用分类模块完成目标的灯质闪烁序列识别,最后采用二元相关性方法将颜色与灯质闪烁序列结合形成航标灯检测结果。检测模块基于YOLOv5模型,改进了其骨干网络并采用了新的SIoU损失函数,实现了视频单帧图像的灯色检测;跟踪模块基于DeepSort,并结合卡尔曼滤波和匈牙利算法实现间歇出现灯色的稳定跟踪;分类模块利用简单的网络结构实现了灯闪频率的快速识别。利用3 500 min的航标灯视频对模型进行训练和测试,结果表明模型的航标灯检测准确率可达90%,检测速度也满足实时性要求。

       

      Abstract: In order to realize the intelligent detection of buoy lights at night, this paper proposes a video-based buoy lights patten detection network.In the network, a detection module is used to capture the color features of buoy lights, then a tracking module is used to track the detected color features and lock light targets, and a classification module is used to further identify the flashing sequence of targets, finally a target′s color and flashing sequence are combined into the detection result of buoy lights by a binary correlation method. The detection module is developed based on YOLOv5, with improvements of its backbone network and loss function(SIoU), realizing light targets′ color detection for video single-frame images. The tracking module is developed based on DeepSort, and Kalman filter and Hungarian algorithm are combined to realize stable tracking of light targets appeared at interval. The classification module uses a simple network structure to quickly identify the flashing frequency of buoy light. 3 500 min videos are used to train and test the network module. The experimental results show that the detection accuracy of the proposed module for nighttime buoy light reaches 90%, and the detection speed also meets real-time requirements.

       

    /

    返回文章
    返回