Multiple attentional path aggregation network for marine object detection

被引:53
作者
Yu, Haifeng [1 ]
Li, Xinbin [1 ]
Feng, Yankai [1 ]
Han, Song [1 ]
机构
[1] Yanshan Univ, Inst Elect Engn, Qinhuangdao 066004, Hebei, Peoples R China
基金
中国国家自然科学基金;
关键词
Marine target detection; Path aggregation network; Multi-attention; Underwater image enhancement; CNN;
D O I
10.1007/s10489-022-03622-0
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Marine target detection is a challenging task because degraded underwater images cause unclear targets. Furthermore, marine targets are small in size and tend to live together. The popular object detection methods perform poorly in marine target detection. Thus, this paper proposes a novel multiple attentional path aggregation network named APAN to improve performance on marine object detection. Firstly, we design a path aggregation network structure which brings features from backbone network to bottom-up path augmentation. Each feature map is enhanced by the lower layer through the bottom-up downsampling pathway and incorporates the features from top-down upsampling layers. Specifically, the last layer fuses feature map from backbone network which enhances the semantic features and improve the ability of feature extraction. Then, a multi-attention which combines coordinate competing attention and spatial supplement attention applies to proposed path aggregation network. Multi-attention can further improve the accuracy of multiple marine object detection. Finally, a double transmission underwater image enhancement algorithm is proposed to enhance the underwater image datasets. The experiments show our method achieves 79.6% mAP in underwater image datasets and 79.03% mAP in enhanced underwater image datasets. Meanwhile, our method achieves 81.5% mAP in PASCAL VOC datasets. In addition, we also applly the method to the underwater robot. The experiments show our method achieves good performance compared with popular object detection methods. The source code is publicly available at https://github.com/yhf2022/APAN.
引用
收藏
页码:2434 / 2451
页数:18
相关论文
共 52 条
[41]   Focal Loss for Dense Object Detection [J].
Lin, Tsung-Yi ;
Goyal, Priya ;
Girshick, Ross ;
He, Kaiming ;
Dollar, Piotr .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :2999-3007
[42]   Automated diabetic retinopathy grading and lesion detection based on the modified R-FCN object-detection algorithm [J].
Wang, Jialiang ;
Luo, Jianxu ;
Liu, Bin ;
Feng, Rui ;
Lu, Lina ;
Zou, Haidong .
IET COMPUTER VISION, 2020, 14 (01) :1-8
[43]   Review on deep learning techniques for marine object recognition: Architectures and algorithms [J].
Wang, Ning ;
Wang, Yuanyuan ;
Er, Meng Joo .
CONTROL ENGINEERING PRACTICE, 2022, 118
[44]   PRI Modulation Recognition Based on Squeeze-and-Excitation Networks [J].
Wei, Shunjun ;
Qu, Qizhe ;
Wu, Yue ;
Wang, Mou ;
Shi, Jun .
IEEE COMMUNICATIONS LETTERS, 2020, 24 (05) :1047-1051
[45]   A transfer convolutional neural network for fault diagnosis based on ResNet-50 [J].
Wen, Long ;
Li, Xinyu ;
Gao, Liang .
NEURAL COMPUTING & APPLICATIONS, 2020, 32 (10) :6111-6124
[46]   Structure Tensor and Guided Filtering-Based Algorithm for Hyperspectral Anomaly Detection [J].
Xie, Weiying ;
Jiang, Tao ;
Li, Yunsong ;
Jia, Xiuping ;
Lei, Jie .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (07) :4218-4230
[47]   Scale-aware feature pyramid architecture for marine object detection [J].
Xu, Fengqiang ;
Wang, Huibing ;
Peng, Jinjia ;
Fu, Xianping .
NEURAL COMPUTING & APPLICATIONS, 2021, 33 (08) :3637-3653
[48]   Multiple attentional pyramid networks for Chinese herbal recognition [J].
Xu, Yingxue ;
Wen, Guihua ;
Hu, Yang ;
Luo, Mingnan ;
Dai, Dan ;
Zhuang, Yishan ;
Hall, Wendy .
PATTERN RECOGNITION, 2021, 110 (110)
[49]   DS-NLCsiNet: Exploiting Non-Local Neural Networks for Massive MIMO CSI Feedback [J].
Yu, Xiaotong ;
Li, Xiangyi ;
Wu, Huaming ;
Bai, Yang .
IEEE COMMUNICATIONS LETTERS, 2020, 24 (12) :2790-2794
[50]   MFR-CNN: Incorporating Multi-Scale Features and Global Information for Traffic Object Detection [J].
Zhang, Hui ;
Wang, Kunfeng ;
Tian, Yonglin ;
Gou, Chao ;
Wang, Fei-Yue .
IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY, 2018, 67 (09) :8019-8030