改进YOLOv8的轻量级光学遥感图像船舶目标检测算法

doi:10.3778/j.issn.1002-8331.2403-0217

摘要/Abstract

摘要： 针对现有基于深度学习的轻量级目标检测算法，在应用于光学遥感图像船舶目标检测任务时所面临的精度低、检测速度慢的情况，提出一种基于YOLOv8s的轻量级光学遥感图像船舶目标检测算法。引入一种新的轻量级非对称检测头，使模型在复杂背景中更加关注船舶对象；主干网络融合选择注意力模块，通过动态调整特征提取主干的感受野来提高目标检测的性能；引入Slim-FPN的思想来改进颈部，在保持检测精度的同时减少参数数量；设计快速卷积模块FasterConv，基于此重构C2f中的Bottleneck结构，命名为Faster_C2f，增强了网络的特征提取能力。实验结果表明，改进的算法在保证检测速度的同时取得了95.2%的检测精度，比基线模型提高1.4%，每秒检测帧数提高8%，模型参数减少33%，较主流算法在检测效果上有一定的提升。

关键词: YOLOv8, 遥感图像, 非对称检测头, 注意力模块, 特征提取

Abstract: Aiming at the low accuracy and slow detection speed faced by existing deep learning-based lightweight target detection algorithms when they are applied to the task of detecting ship targets in optical remote sensing images, a lightweight ship target detection algorithm based on YOLOv8s is proposed for optical remote sensing images. A new lightweight asymmetric detection head is introduced to make the model pay more attention to ship targets in complex backgrounds. The backbone network fusion selects an attention module to improve the performance of target detection by dynamically adjusting the sensing field of the feature extraction backbone. The idea of Slim-FPN is introduced to improve the neck part, which reduces the number of parameters while maintaining the detection accuracy. A new fast convolutional module FasterConv is designed, based on which, the bottleneck structure in C2f is reconstructed and named Faster_C2f, which enhances the feature extraction ability of the network. The experimental results show that the improved algorithm achieves a detection accuracy of 95.2% while ensuring the detection speed, which is 1.4% higher than the baseline model, the number of detected frames per second is increased by 8%, and the model parameters are reduced by 33%, which is a certain improvement over the mainstream algorithms in terms of detection effect.

Key words: YOLOv8, remote sensing image, asymmetric detection head, attention module, feature extraction

杨志渊, 罗亮, 吴天阳, 于博向. 改进YOLOv8的轻量级光学遥感图像船舶目标检测算法[J]. 计算机工程与应用, 2024, 60(16): 248-257.

YANG Zhiyuan, LUO Liang, WU Tianyang, YU Boxiang. Improved Lightweight Ship Target Detection Algorithm for Optical Remote Sensing Images with YOLOv8[J]. Computer Engineering and Applications, 2024, 60(16): 248-257.

参考文献

[1] CAI C, CHEN L, ZHANG X, et al. End-to-end optimized ROI image compression[J]. IEEE Transactions on Image Processing, 2020, 29: 3442-3457.
[2] REN S, HE K, GIRSHICK R, et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39(6): 1137-1149.
[3] GE Z, LIU S, WANG F, et al. YOLOx: exceeding YOLO series in 2021[J]. arXiv:2107.08430, 2021.
[4] REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified, real-time object detection[C]//Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition, 2016: 779-788.
[5] REDMON J, FARHADI A. YOLO9000: better, faster, stronger[C]//Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition, 2017: 6517-6525.
[6] REDMON J, FARHADI A. YOLOv3: an incremental improvement[J]. arXiv:1804.02767, 2018.
[7] LIU W, ANGUELOV D, ERHAN D, et al. SSD: single shot multibox detector[J]. arXiv:1512.02325, 2015.
[8] 沈丰毅, 王琴, 蒋剑飞. 基于改进卷积神经网络舰船的实时目标检测算法[J]. 计算机应用研究, 2020, 37(S1): 316-319.
SHENG F Y, WANG Q, JIANG J F. Real-time target detection algorithm for ships based on improved convolutional neural network[J]. Application Research of Computers, 2020, 37(S1): 316-319.
[9] 王欣, 江涛, 马珍, 等. 基于改进YOLOv3-Tiny的遥感舰船小目标检测方法 [J]. 计算机时代, 2023(3): 111-115.
WANG X, JIANG T, MA Z, et al. Small target detection of remote sensing ship based on improved YOLOv3-Tiny algorithm[J].Computer Era, 2023(3): 111-115.
[10] 成倩, 李佳, 杜娟. 基于YOLOv5的光学遥感图像舰船目标检测算法[J]. 系统工程与电子技术, 2023, 45(5): 1270-1276.
CHENG Q, LI J, DU J. Ship target detection algorithm of optical remote sensing image based on YOLOv5[J]. Systems Engineering and Electronics, 2023, 45(5): 1270-1276.
[11] 郎磊, 刘宽, 王东. 基于YOLOX-Tiny的轻量级遥感图像目标检测模型[J]. 激光与光电子学进展, 2023, 60(2): 353-363.
LANG L, LIU K, WANG D. Lightweight remote sensing object detector based on YOLOX-Tiny[J]. Laser & Optoelectronics Progress, 2023, 60(2): 353-363.
[12] ZHANG J, CHEN Z, YAN G, et al. Faster and lightweight: an improved YOLOv5 object detector for remote sensing images [J]. Remote Sensing, 2023, 15(20): 4974.
[13] LI Y, HOU Q, ZHENG Z, et al. Large selective kernel network for remote sensing object detection[J]. arXiv:2303. 09030, 2023.
[14] ZHANG M, GAO F, YANG W, et al. Wildlife object detection method applying segmentation gradient flow and feature dimensionality reduction[J]. Electronics, 2023, 12(2): 377.
[15] CHEN J, KAO S-H, HE H, et al. Run, don’t walk: chasing higher FLOPS for faster neural networks[C]//Proceedings of the 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023: 12021-12031.
[16] WOO S, PARK J, LEE J-Y, et al. CBAM: convolutional block attention module [C]//Proceedings of the 15th European Conference on Computer Vision, 2018: 3-19.
[17] YANG B, BENDER G, LE Q V, et al. CondConv: conditionally parameterized convolutions for efficient inference[C]// Advances in Neural Information Processing Systems 32, 2019.
[18] CHEN Y, DAI X, LIU M, et al. Dynamic convolution: attention over convolution kernels[C]//Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2020: 11030-11039.
[19] LI X, WANG W, HU X, et al. Selective kernel networks [C]//Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2019: 510-519.
[20] ZHU L, WANG X, KE Z, et al. BiFormer: vision transformer with bi-level routing attention[J] arXiv:2303.08810, 2023.
[21] OUYANG D, HE S, ZHANG G, et al. Efficient multi-scale attention module with cross-spatial learning[J]. arXiv:2305. 13563, 2023.
[22] HAN J, DING J, XUE N, et al. ReDet: a rotation-equivariant detector for aerial object detection[J]. arXiv:2103.07733,2021.
[23] XU Y, FU M, WANG Q, et al. Gliding vertex on the horizontal bounding box for multi-oriented object detection[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2021, 43(4): 1452-1459.
[24] DING J, XUE N, LONG Y, et al. Learning RoI transformer for oriented object detection in aerial images[C]//Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2020.
[25] XIAORAN W. Real-time oriented ship detection in aerial images for embedded platforms[C]//Proceedings of the 2022 19th International Computer Conference on Wavelet Active Media Technology and Information Processing, 2022: 1-5.
[26] YANG J, WU B, ZHANG H, et al. Rotating frame labeling and improved YOLOv7-based algorithm for vehicle object detection[C]//Proceedings of the 2023 IEEE 6th International Conference on Pattern Recognition and Artificial Intelligence, 2023: 170-175.