张炳焱, 张闯, 石振男, 刘松涛. 基于YOLO-FNC模型的轻量化船舶检测方法[J]. 中国舰船研究. DOI: 10.19693/j.issn.1673-3185.03487
引用本文: 张炳焱, 张闯, 石振男, 刘松涛. 基于YOLO-FNC模型的轻量化船舶检测方法[J]. 中国舰船研究. DOI: 10.19693/j.issn.1673-3185.03487
Lightweight ship detection method based on YOLO-FNC model[J]. Chinese Journal of Ship Research. DOI: 10.19693/j.issn.1673-3185.03487
Citation: Lightweight ship detection method based on YOLO-FNC model[J]. Chinese Journal of Ship Research. DOI: 10.19693/j.issn.1673-3185.03487

基于YOLO-FNC模型的轻量化船舶检测方法

Lightweight ship detection method based on YOLO-FNC model

  • 摘要: 针对交通密集的港口、船舶聚集的渔船作业区以及船岸混合交通场景等复杂环境,提出了一种基于YOLO-FNC的轻量化且高效船舶检测方法。首先,设计一种基于FasterNet思想的神经网络模块FasterNext,并将该模块替换YOLO模型中的C3模块,在不影响准确性的条件下确保运行速度更快。其次,将NAM注意力机制融入到网络结构中,通过利用稀疏的权重惩罚抑制特征权重确保权重的计算更加高效。最后,提出新的边界框回归损失以加快预测帧调整并增加帧回归率,提升网络模型收敛速度。实验结果表明,在自建的复杂场景下船舶数据集进行检测实验,与YOLOv5s算法相比,提出方法的mAP@0.5提升了6.3%,参数量减少了9.74%,计算量减少了11.4%,有效地实现了轻量化、高精度的船舶检测。

     

    Abstract: A lightweight and efficient ship detection method based on YOLO-FNC was proposed for the complex environment such as the port with dense traffic. First, a neural network module FasterNext based on the FasterNet method is designed, and this module replaces the C3 module in the YOLO model to ensure faster operation without affecting the accuracy. Second, the NAM(Normalization-based Attention Module) is integrated into the network structure, and the sparse weight penalty is used to suppress the feature weights to ensure more efficient weight calculation. Finally, a new bounding box regression loss is proposed to speed up the prediction frame adjustment and increase the regression rate to improve the rate of convergence of the network mode. The experimental results show that the proposed method performs detection experiments on ship datasets in a self built complex scene, in which improve mAP@0.5 by 6.3%, reduce parameter count by 9.74%, and reduce computational complexity by 11.4%, effectively achieving lightweight and high-precision ship detection compared with the YOLOv5s algorithm.

     

/

返回文章
返回