Citation: | LI Chenxuan, GU Jiaojiao, WANG Lei, et al. Warship's vital parts detection algorithm based on lightweight Anchor-Free network with multi-scale feature fusion[J]. Journal of Beijing University of Aeronautics and Astronautics, 2022, 48(10): 2006-2019. doi: 10.13700/j.bh.1001-5965.2021.0050(in Chinese) |
One of the key technologies of precision-guidance weapons is the anti-ship missile's ability to strike vital parts of a warship with pinpoint accuracy. Aiming at the problems of low detection accuracy, insu-fficient ability in feature extraction and the processing of the generated-anchors reduces the detection speed in anti-ship missile seekers, a warship's vital parts detection algorithm based on a lightweight Anchor-Free network with multi-scale feature fusion is proposed. Due to the multi-scale and multi-angle characteristics of the vital parts detection data, the multi-scale feature fusion module is introduced to optimize the feature extraction by comprehensively using the detection information of different receptive fields. To boost the detection accuracy and reduce the total parameters of the algorithm, the skip connections in Hourglass are enhanced by using the efficient and lightweight attention mechanism. The transfer-learning is used to improve the convergence of this algorithm effectively. Experiments were carried out on the dataset of the warship's vital parts and the PASCAL VOC. Experimental results show the mAP is increased by 4.41% and 5.57% respectively. The algorithm's parameters and the computation are analyzed. The module ablation experiments are designed to demonstrate the effectiveness of the algorithm.
[1] |
QI S, MA J, LIN J, et al. Unsupervised ship detection based on saliency and S-HOG descriptor from optical satellite images[J]. Geoscience and Remote Sensing Letters, 2015, 12(7): 1451-1455. doi: 10.1109/LGRS.2015.2408355
|
[2] |
黄洁, 姜志国, 张浩鹏, 等. 基于卷积神经网络的遥感图像舰船目标检测[J]. 北京航空航天大学学报, 2017, 43(9): 1841-1848. doi: 10.13700/j.bh.1001-5965.2016.0755
HUANG J, JIANG Z G, ZHANG H P, et al. Ship object detection in remote sensing images using convolutional neural networks[J]. Journal of Beijing University of Aeronautics and Astronautics, 2017, 43(9): 1841-1848(in Chinese). doi: 10.13700/j.bh.1001-5965.2016.0755
|
[3] |
REN S Q, HE K M, 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. doi: 10.1109/TPAMI.2016.2577031
|
[4] |
李健伟, 曲长文, 彭书娟, 等. 基于卷积神经网络的SAR图像舰船目标检测[J]. 系统工程与电子技术, 2018, 40(9): 1953-1959. https://www.cnki.com.cn/Article/CJFDTOTAL-XTYD201809009.htm
LI J W, QU C W, PENG S J, et al. Ship detection in SAR images based on convolutional neural network[J]. Systems Engineering and Electronics, 2018, 40(9): 1953-1959(in Chinese). https://www.cnki.com.cn/Article/CJFDTOTAL-XTYD201809009.htm
|
[5] |
张旭. 基于深度学习的双阶段舰船检测算法研究[D]. 南京: 东南大学, 2019: 35-40.
ZHANG X. Research on two-state ship detection method based on deep learning[D]. Nanjing: Southeast University, 2019: 35-40(in Chinese).
|
[6] |
GAO L R, HE Y Q, SUN X, et al. Incorporating negative sample training for ship detection based on deep learning[J]. Sensors, 2019, 19(3): 684-704. doi: 10.3390/s19030684
|
[7] |
王玺坤, 姜宏旭, 林珂玉. 基于改进型YOLO算法的遥感图像舰船检测[J]. 北京航空航天大学学报, 2020, 46(6): 1184-1191. doi: 10.13700/j.bh.1001-5965.2019.0394
WANG X K, JIANG H X, LIN K Y. Remote sensing image ship detection based on modified YOLO algorithm[J]. Journal of Beijing University of Aeronautics and Astronautics, 2020, 46(6): 1184-1191(in Chinese). doi: 10.13700/j.bh.1001-5965.2019.0394
|
[8] |
WANG Y Y, WANG C, ZHANG H, et al. Automatic ship detection based on RetinaNet using multi-resolution Gaofen-3 imagery[J]. Remote Sensing, 2019, 11(5): 531-545. doi: 10.3390/rs11050531
|
[9] |
杨龙, 苏娟, 李响. 基于深度卷积神经网络的SAR舰船目标检测[J]. 系统工程与电子技术, 2019, 41(9): 1990-1997. https://www.cnki.com.cn/Article/CJFDTOTAL-XTYD201909011.htm
YANG L, SU J, LI X. Ship detection in SAR images based on deep convolutional neural network[J]. Systems Engineering and Electronics, 2019, 41(9): 1990-1997(in Chinese). https://www.cnki.com.cn/Article/CJFDTOTAL-XTYD201909011.htm
|
[10] |
郑婷婷, 杨雪, 戴阳. 基于关键点的Anchor-Free目标检测模型综述[J]. 计算机系统应用, 2020, 29(8): 1-8. https://www.cnki.com.cn/Article/CJFDTOTAL-XTYY202008002.htm
ZHENG T T, YANG X, DAI Y. Overview of Anchor-Free object detection model based on key points[J]. Computer Systems and Applications, 2020, 29(8): 1-8(in Chinese). https://www.cnki.com.cn/Article/CJFDTOTAL-XTYY202008002.htm
|
[11] |
TIAN Z, SHEN C H, CHEN H, et al. FCOS: Fully convolutional one-stage object detection[C]//Proceedings of the Conference on International Conference on Computer Vision. Piscataway: IEEE Press, 2019: 1-13.
|
[12] |
ZHOU X Y, WANG D Q, PHILIPP K, et al. Objects as points[EB/OL]. (2019-04-25)[2021-01-20]. https://arxiv.org/abs/1904.07850.
|
[13] |
HOU Q B, ZHANG L, CHENG M M, et al. Strip pooling: Rethinking spatial pooling for scene parsing[C]//Proceedings of the Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE Press, 2020: 4002-4011.
|
[14] |
XIAO B, WU H P, WEI Y C. Simple baselines for human pose estimation and tracking[C]//Proceedings of the Conference on European Conference on Computer Vision. Berlin: Springer, 2018: 472-487.
|
[15] |
YU F, WANG D, SHELHAMER E, et al. Deep layer aggregation[C]//Proceedings of the Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE Press, 2018: 2403-2412.
|
[16] |
WANG Q L, WU B G, ZHU P F, et al. Efficient channel attention for deep convolutional neural networks[C]//Proceedings of the Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE Press, 2019: 1-12.
|
[17] |
王新, 李喆, 张宏立. 一种迭代聚合的高分辨率网络Anchor-free目标检测方法[J]. 北京航空航天大学学报, 2021, 47(12): 2533-2541. doi: 10.13700/j.bh.1001-5965.2020.0484
WANG X, LI Z, ZHANG H L. High resolution network Anchor-free object detection method based on iterative aggregation[J]. Journal of Beijing University of Aeronautics and Astronautics, 2021, 47(12): 2533-2541(in Chinese). doi: 10.13700/j.bh.1001-5965.2020.0484
|
[18] |
HU J, SHEN L, ALBANIE S, et al. Squeeze-and-excitation networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2020, 42(8): 2011-2023.
|
[19] |
张雪松, 庄严, 闫飞, 等. 基于迁移学习的类别级物体识别与检测研究与进展[J]. 自动化学报, 2019, 45(7): 1224-1243. https://www.cnki.com.cn/Article/CJFDTOTAL-MOTO201907002.htm
ZHANG X S, ZHUANG Y, YAN F, et al. Status and development of transfer learning based category-level object recognition and detection[J]. Acta Automatica Sinica, 2019, 45(7): 1224-1243(in Chinese). https://www.cnki.com.cn/Article/CJFDTOTAL-MOTO201907002.htm
|
[20] |
WANG Y Y, WANG C, ZHANG H. Combining a single shot multibox detector with transfer learning for ship detection using sentinel-1 SAR images[J]. Remote Sensing Letters, 2018, 9(8): 780-788.
|
[21] |
BRIJRAJ S, DURGA T, KUMAR A S. Shunt connection: An intelligent skipping of contiguous blocks for optimizing MobileNet-V2[J]. Neural Networks, 2019, 118: 192-203.
|