|
|
Underwater Target Detection Algorithm Based on Improved FCOS |
CHEN Wei-dong1,2,XIE Xiao-dong1,CEN Qiang1,CHEN Na-lan1,ZHU Qi-guang1,2 |
1. School of Information Science and Engineering, Yanshan University, Qinhuangdao, Hebei 066004, China
2. Key Laboratory for Special Fiber and Fiber Sensor of Hebei Province, Yanshan University, Qinhuangdao, Hebei 066004, China |
|
|
Abstract Propose an underwater target detection algorithm based on an improved fully convolutional one-stage object detection (FCOS). In response to problems such as high color cast, low contrast, dark color bias, and blurry distortion in underwater optical images that result in poor performance of existing target detection algorithms in underwater environments, the ordinary convolutions in the backbone network are replaced with deformable convolutions (DCN) for optimization, enhancing the algorithm's feature extraction capability in blurry underwater images. The feature fusion network and detection network of the improved network are enhanced through neural architecture search (NAS) to improve the utilization of features extracted by the backbone network. The CIoU Loss is adopted as the new loss function to improve the accuracy of coordinate regression. Experimental results show that the algorithm improves the detection accuracy by 1.8% and recall rate by 2.2% on the DUO dataset, with a detection speed of 53.4 frames /s, which is a 5.0% reduction compared to the previous version. The algorithm achieves high accuracy and meets the requirements of real-time detection.
|
Received: 16 February 2023
Published: 17 November 2023
|
|
Fund:National Natural Science Foundation |
|
|
|
[6] |
Sung M S, Yu S C, Girdhar Y. Vision based real-time fish detection using convolutional neural network[C]//OCEANS 2017-Aberdeen. Aberdeen, UK, 2017: 1-6.
|
[15] |
高美静, 李时雨, 刘泽昊, 等. 基于改进Faster R-CNN的水母检测与识别算法[J]. 计量学报, 2023, 44(1): 54-61.
|
[3] |
李鹤喜, 李记花, 李威龙. 一种基于注意机制和卷积神经网络的视觉模型[J]. 计量学报, 2021, 42(7): 840-845.
|
[4] |
Li X, Shang M, Hao J, et al. Accelerating fish detection and recognition by sharing CNNs with objectness learning[C]// OCEANS 2016-Shanghai. Shanghai, China, 2016: 1-5.
|
[8] |
Chen L Y, Zheng M C, Duan S Q, et al. Underwater target recognition based on improved YOLOv4 neural network[J]. Electronics, 2021, 10(14): 1634.
|
[1] |
Almabouada F, Abreu M A, Coelho J M P, et al. Experimental and simulation assessments of underwater light propagation[J]. Frontiers of Optoelectronics, 2019, 12: 405-412.
|
[7] |
Cai K W, Miao X Y, Wang W, et al. A modified YOLOv3 model for fish detection based on MobileNetv1 as backbone[J]. Aquacultural Engineering, 2020, 91: 102117.
|
[10] |
Liu H, Song P H, Ding R W. Towards domain generalization in underwater object detection[C]//2020 IEEE International Conference on Image Processing (ICIP). Abu Dhabi, United Arab Emirates, 2020: 1971-1975.
|
[11] |
Zeng L C, Sun B, Zhu D Q. Underwater target detection based on Faster R-CNN and adversarial occlusion network[J]. Engineering Applications of Artificial Intelligence, 2021, 100: 104190.
|
[12] |
Lin W H, Zhong J X, Liu S, et al. RoIMix: proposal-fusion among multiple images for underwater object detection[C]//ICASSP 2020-2020 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). Barcelona, Spain, 2020: 2588-2592.
|
[14] |
Peng F, Miao Z, Li F, et al. S-FPN: A shortcut feature pyramid network for sea cucumber detection in underwater images[J]. Expert Systems with Applications, 2021, 182: 115306.
|
|
Gao M J, Li S Y, Liu H Z, et al. Jellyfish Detection and Recognition Algorithm Based on lmproved Faster R-CNN[J]. Acta Metrologica Sinica, 2023, 44(1): 54-61.
|
[17] |
Zhu X Z, Hu H, Lin S, et al. Deformable convnets v2: More deformable, better results[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Long Beach, USA, 2019: 9308-9316.
|
[19] |
Wang N, Gao Y, Chen H, et al. NAS-FCOS: Fast neural architecture search for object detection[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle, USA, 2020: 11943-11951.
|
|
Li H X, Li J H, Li W H. A Visual Model Based on Attention Mechanism and Convolutional Neural Network[J]. Acta Metrologica Sinica, 2021, 42(7): 840-845.
|
[5] |
Zhang J, Zhu L, Xu L, et al. MFFSSD: An Enhanced SSD for Underwater Object Detection[C]//2020 Chinese Automation Congress (CAC). Shanghai, China, 2020: 5938-5943.
|
[9] |
Yeh C H, Lin C H, Kang L W, et al. Lightweight deep neural network for joint learning of underwater object detection and color conversion[J]. IEEE Transactions on Neural Networks and Learning Systems, 2021, 33(11): 6129-6143.
|
[13] |
Chen L Y, Liu Z, Tong L, et al. Underwater object detection using Invert Multi-Class Adaboost with deep learning[C]//2020 International Joint Conference on Neural Networks (IJCNN). IEEE, 2020: 1-8.
|
[20] |
Zheng Z H, Wang P, Liu W, et al. Distance-IoU loss: Faster and better learning for bounding box regression[C]//Proceedings of the AAAI Conference on Artificial Intelligence. New York, USA, 2020, 34(7): 12993-13000.
|
[2] |
程淑红, 马继勇, 张仕军, 等. 改进的混合高斯与YOLOv2融合烟雾检测算法[J]. 计量学报, 2019, 40(5): 798-803.
|
|
Cheng S H, Ma J Y, Zhang S J, et al. Smoke Detection Algorithm Combined with lmproved Gaussian Mixture and YOLOv2[J]. Acta Metrologica Sinica, 2019, 40(5): 798-803.
|
[16] |
Liu C W, Li H J, Wang S C, et al. A dataset and benchmark of underwater object detection for robot picking[C]//2021 IEEE International Conference on Multimedia & Expo Workshops (ICMEW). Shenzhen, China, 2021: 1-6.
|
[18] |
Ghiasi G, Lin T Y, Le Q V. Nas-fpn: Learning scalable feature pyramid architecture for object detection[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Long Beach, USA, 2019: 7036-7045.
|
|
|
|