RHS-YOLOv8: A Lightweight Underwater Small Object Detection Algorithm Based on Improved YOLOv8
Source abstract
Abstract
To address the challenge posed by the abundance of small objects with weak object features and little information in the images of underwater biomonitoring scenarios, and the added difficulty of recognizing these objects due to light absorption and scattering in the underwater environment, this study proposes an improved RHS-YOLOv8 (Ref-Dilated-HBFPN-SOB-YOLOv8). Firstly, a combination of hybrid inflated convolution and RefConv is used to redesign the lightweight Ref-Dilated convolution block, which reduces the model computation. Second, a new feature pyramid network fusion module, the Hybrid Bridge Feature Pyramid Network (HBFPN), is designed to fuse the deep features with the high-level features, as well as the features of the current layer, to improve the feature extraction capability for fuzzy objects. Third, Efficient Localization Attention (ELA) is added to reduce the interference of irrelevant factors on prediction. Fourth, an Involution module is introduced to effectively capture spatial long-range relationships and improve recognition accuracy. Finally, a small object detection branch is incorporated into the original architecture to enhance the model’s performance in detecting small objects. Experiments based on the DUO dataset show that RHS-YOLOv8 reduces 9.95% of computing power, while mAP@0.5 and mAP@0.50:0.95 are improved by 2.54% and 4.31%, respectively. Compared with other cutting-edge underwater object detection algorithms, the present algorithm improves the detection accuracy while lightweighting the improvement, which effectively enhances the capability to detect small underwater objects.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Advanced Neural Network Applications · Image Enhancement Techniques · Water Quality Monitoring Technologies
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
College of Artificial Intelligence, Jianghan University, Wuhan 430056, China
Funding
Postgraduate Cultivating Fund of Jianghan University · Innovation Graduate Research Fund of Jianghan University
References (49)
Show source-provided references
1. Vision-based object detection and tracking for autonomous navigation of underwater robots10.1016/j.oceaneng.2012.04.006 ↗
2. A systematic review and analysis of deep learning-based underwater object detection10.1016/j.neucom.2023.01.056 ↗
3. Underwater hyperspectral imaging: A new tool for marine archaeology10.1364/AO.57.003214 ↗
4. Real-time detection of uneaten feed pellets in underwater images for aquaculture using an improved YOLO-V4 network10.1016/j.compag.2021.106135 ↗
5. Real-time underwater object detection based on DC resistivity method10.1109/TGRS.2016.2591619 ↗
6. Controlling waves in space and time for imaging and focusing in complex media10.1038/nphoton.2012.88 ↗
7. Underwater image enhancement by wavelength compensation and dehazing10.1109/TIP.2011.2179666 ↗
8. Do marine substrates ‘look’ and ‘sound’ the same? Supervised classification of multibeam acoustic data using autonomous underwater vehicle images10.1016/j.ecss.2012.11.001 ↗
9. Inspection and monitoring systems subsea pipelines: A review paper10.1177/1475921719837718 ↗
10. Kim, D., Lee, D., Myung, H., and Choi, H.T. (2012, January 21–24). Object detection and tracking for autonomous underwater robots using weighted template matching. Proceedings of the 2012 Oceans-Yeosu, Yeosu, Republic of Korea.10.1109/OCEANS-Yeosu.2012.6263501 ↗
11. Saini, A., and Biswas, M. (2019, January 23–25). Object detection in underwater image by detecting edges using adaptive thresholding. Proceedings of the 2019 3rd International Conference on Trends in Electronics and Informatics (ICOEI), Tirunelveli, India.10.1109/ICOEI.2019.8862794 ↗
12. Object detection in underwater acoustic images using edge based segmentation method10.1016/j.procs.2020.01.015 ↗
13. Two-branch deep neural network for underwater image enhancement in HSV color space10.1109/LSP.2021.3099746 ↗
14. Girshick, R. (2015, January 7–13). Fast r-cnn. Proceedings of the IEEE International Conference on Computer Vision 2015, Santiago, Chile.10.1109/ICCV.2015.169 ↗
15. Ren, S., He, K., Girshick, R., and Sun, J. (2015, January 7–12). Faster r-cnn: Towards real-time object detection with region proposal networks. Proceedings of the Advances in Neural Information Processing Systems 2015, Montreal, QC, Canada.
16. He, K., Gkioxari, G., Dollár, P., and Girshick, R. (2017, January 22–29). Mask r-cnn. Proceedings of the IEEE International Conference on Computer Vision 2017, Venice, Italy.10.1109/ICCV.2017.322 ↗
17. An evaluation of retinanet on indoor object detection for blind and visually impaired persons assistance navigation10.1007/s11063-020-10197-9 ↗
18. Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C.Y., and Berg, A.C. (2016, January 11–14). Ssd: Single shot multibox detector. Proceedings of the Computer Vision–ECCV 2016: 14th European Conference, Amsterdam, The Netherlands. Proceedings, Part I 14.10.1007/978-3-319-46448-0_2 ↗
19. Saliency-guided single shot multibox detector for object detection in SAR images10.1109/TGRS.2019.2953936 ↗
20. Redmon, J., Divvala, S., Girshick, R., and Farhadi, A. (2016, January 27–30). You only look once: Unified, real-time object detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition 2016, Las Vegas, NV, USA.10.1109/CVPR.2016.91 ↗
21. Redmon, J., and Farhadi, A. (2018). Yolov3: An incremental improvement. arXiv.
22. Wu, W., Liu, H., Li, L., Long, Y., Wang, X., Wang, Z., Li, J., and Chang, Y. (2021). Application of local fully Convolutional Neural Network combined with YOLO v5 algorithm in small object detection of remote sensing image. PLoS ONE, 16.10.1371/journal.pone.0259283 ↗
23. Hussain, M. (2023). YOLO-v1 to YOLO-v8, the rise of YOLO and its complementary nature toward digital manufacturing and industrial defect detection. Machines, 11.10.3390/machines11070677 ↗
24. Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C.Y., and Berg, A.C. (2016, January 11–14). SSD: Single Shot MultiBox Detector. Proceedings of the European Conference on Computer Vision (ECCV), Amsterdam, The Netherlands.10.1007/978-3-319-46448-0_2 ↗
25. Lin, T.Y., Goyal, P., Girshick, R., He, K., and Dollár, P. (2017, January 22–29). Focal Loss for Dense Object Detection. Proceedings of the IEEE International Conference on Computer Vision 2017, Venice, Italy.10.1109/ICCV.2017.324 ↗
26. Carion, N., Massa, F., Synnaeve, G., Usunier, N., Kirillov, A., and Zagoruyko, S. (2020, January 23–28). End-to-End Object Detection with Transformers. Proceedings of the European Conference on Computer Vision (ECCV), Glasgow, UK.10.1007/978-3-030-58452-8_13 ↗
27. Lei, F., Tang, F., and Li, S. (2022). Underwater target detection algorithm based on improved YOLOv5. J. Mar. Sci. Eng., 10.10.3390/jmse10030310 ↗
28. Yan, J., Zhou, Z., Zhou, D., Su, B., Xuanyuan, Z., Tang, J., Lai, Y., Chen, J., and Liang, W. (2022). Underwater object detection algorithm based on attention mechanism and cross-stage partial fast spatial pyramidal pooling. Front. Mar. Sci., 9.10.3389/fmars.2022.1056300 ↗
29. Woo, S., Park, J., Lee, J.Y., and Kweon, I.S. (2018, January 8–14). CBAM: Convolutional block attention module. Proceedings of the European Conference on Computer Vision 2018, Munich, Germany.10.1007/978-3-030-01234-2_1 ↗
30. Zhang, M., Wang, Z., Song, W., Zhao, D., and Zhao, H. (2024). Efficient Small-Object Detection in Underwater Images Using the Enhanced YOLOv8 Network. Appl. Sci., 14.10.3390/app14031095 ↗
31. Chen, J., Kao, S.H., He, H., Zhuo, W., Wen, S., Lee, C.H., and Chan, S.H. (2023, January 17–24). Run, don’t walk: Chasing higher FLOPS for faster neural networks. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition 2023, Vancouver, BC, Canada.10.1109/CVPR52729.2023.01157 ↗
32. Srinivas, A., Lin, T.Y., Parmar, N., Shlens, J., Abbeel, P., and Vaswani, A. (2021, January 19–25). Bottleneck transformers for visual recognition. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition 2021, Nashville, TN, USA.10.1109/CVPR46437.2021.01625 ↗
33. Ultralytics (2024, May 24). Track and Count Objects Using YOLOv8. Roboflow. Available online: https://blog.roboflow.com/yolov8-tracking-and-counting/#object-detection-with-yolov8.
34. Ge, Z., Liu, S., Wang, F., Li, Z., and Sun, J. (2021). Yolox: Exceeding yolo series in 2021. arXiv.
35. Yu, F. (2015). Multi-scale context aggregation by dilated convolutions. arXiv.
36. Cai, Z., Ding, X., Shen, Q., and Cao, X. (2023). Refconv: Re-parameterized refocusing convolution for powerful convnets. arXiv.
37. Wang, C.Y., Yeh, I.H., and Liao, H.Y.M. (2024). Yolov9: Learning what you want to learn using programmable gradient information. arXiv.10.1007/978-3-031-72751-1_1 ↗
38. Liu, W., Lu, H., Fu, H., and Cao, Z. (2023, January 2–3). Learning to upsample by learning to sample. Proceedings of the IEEE/CVF International Conference on Computer Vision 2023, Paris, France.10.1109/ICCV51070.2023.00554 ↗
39. Xu, W., and Wan, Y. (2024). ELA: Efficient Local Attention for Deep Convolutional Neural Networks. arXiv.
40. Hou, Q., Zhou, D., and Feng, J. (2021, January 19–25). Coordinate attention for efficient mobile network design. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, TN, USA.10.1109/CVPR46437.2021.01350 ↗
41. Wu, Y., and He, K. (2018, January 8–14). Group normalization. Proceedings of the European Conference on Computer Vision 2018, Munich, Germany.10.1007/978-3-030-01261-8_1 ↗
42. Li, D., Hu, J., Wang, C., Li, X., She, Q., Zhu, L., Zhang, T., and Chen, Q. (2021, January 19–25). Involution: Inverting the inherence of convolution for visual recognition. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition 2021, Nashville, TN, USA.10.1109/CVPR46437.2021.01214 ↗
43. Liu, C., Li, H., Wang, S., Zhu, M., Wang, D., Fan, X., and Wang, Z. (2021, January 5–9). A dataset and benchmark of underwater object detection for robot picking. Proceedings of the 2021 IEEE International Conference on Multimedia & Expo Workshops (ICMEW), Shenzhen, China.10.1109/ICMEW53276.2021.9455997 ↗
44. Rethinking general underwater object detection: Datasets, challenges, and solutions10.1016/j.neucom.2022.10.039 ↗
45. AWF-YOLO: Enhanced Underwater Object Detection with Adaptive Weighted Feature Pyramid Network10.20517/ces.2023.19 ↗
46. Zhao, Y., Sun, F., and Wu, X. (2024). FEB-YOLOv8: A multi-scale lightweight detection model for underwater object detection. PLoS ONE, 19.10.1371/journal.pone.0311173 ↗
47. CEH-YOLO: A composite enhanced YOLO-based model for underwater object detection10.1016/j.ecoinf.2024.102758 ↗
48. Dynamic YOLO for small underwater object detection10.1007/s10462-024-10788-1 ↗
49. Yang, C., Xiang, J., Li, X., and Xie, Y. (2024). FishDet-YOLO: Enhanced underwater fish detection with richer gradient flow and long-range dependency capture through Mamba-C2f. Electronics, 13.10.3390/electronics13183780 ↗