Underwater Trash Detection Method Based on Improved YOLOv5

doi:10.3969/j.issn.1006-2475.2024.07.018

Abstract

Abstract: To address the limitations of underwater image acquisition such as insufficient light， high noise and unclear object recognition， which lead to the ineffectiveness of existing object detection algorithms， an underwater garbage object detection algorithm based on improved YOLOv5 is proposed. The purpose of the improved object detection algorithm is to achieve more accurate detection and removal of underwater plastic trash from the ocean. The improved algorithm containes some improvements：using the Contrast Limited Adaptive Histogram Equalization（CLAHE） algorithm to enhance data features， which reduces the difficulty of feature extraction and enables the network to be detected more flexibly and more accurately； introducing a parameter-free attention module SimAM， using the lightweight convolution method GSConv to enhance network extraction capability while reducing model computation; At the same time， multi-scale feature fusion detection is added to solve the problem of small target location of underwater debris. Numbers of experiments are conducted based on MarineTrash which is a self-built real underwater environmental litter dataset， the results show that the improved method has good performance， in which the accuracy is increased by 4.3 percentage points， the mAP is increased by 3.5 percentage points， the GFLOPs is reduced by 0.3， and the model weight is only 13.9 MB， which is 0.6 MB lower than the baseline. The research on the underwater trash detection algorithm based on the improved YOLOv5 provides sufficient technology for deploying and installing detectors in Autonomous Underwater Vehicles （AUVs） to achieve detection and automatic removal of marine underwater trash and maintain the marine ecosystem.

Key words: , object detection, underwater trash, multi-scale feature fusion, YOLOv5, GSConv, SimAM

CLC Number:

TP391

PANG Mei, WANG Gong, ZHAN Yong, HUANG Zhefa. Underwater Trash Detection Method Based on Improved YOLOv5[J]. Computer and Modernization, 2024, 0(07): 120-126.

References

［1］ MAJCHROWSKA S， MIKOŁAJCZYK A， FERLIN M， et al. Waste detection in Pomerania： Non-profit project for detecting waste in environment［J］. arXiv preprint arXiv：2105.06808， 2021.
［2］ WANG J Y， YU N G. UTD-YOLOv5： A real-time underwater targets detection method based on attention improved YOLOv5［J］. arXiv preprint arXiv：2207.00837， 2022.
［3］ DAKHIL R A， KHAYEAT A R H. Review on deep learning technique for underwater object detection［J］. arXiv preprint arXiv：2209.10151， 2022.
［4］ FULTON M， HONG J S， ISLAM M J， et al. Robotic detection of marine litter using deep visual detection models［C］// 2019 International Conference on Robotics and Automation （ICRA）. IEEE， 2019：5752-5758.
［5］ ZHANG Y X， HUANG J， CAI F H. On bridge surface crack detection based on an improved YOLOv3 algorithm［J］. IFAC-PapersOnLine， 2020，53（2）：8205-8210.
［6］李世伟. 基于YOLOv5算法的目标检测与车牌识别系统［J］. 电子技术与软件工程， 2022（1）：138-141.
［7］陈宇梁，董绍江，孙世政，等. 改进YOLOv5的弱光水下生物目标检测算法［J］. 北京航空航天大学学报， 2024，50（2）：499-507.
［8］ CHEN X L， GUPTA A. An implementation of Faster RCNN with study for region sampling［J］. arXiv preprint arXiv：1702.02138， 2017.
［9］ REDMON J， FARHADI A. YOLOv3： An incremental improvement［J］. arXiv preprint arXiv：1804.02767， 2018.
［10］ KREMEZI M， KRISTOLLARI V， KARATHANASSI V， et al. Pansharpening PRISMA data for marine plastic litter detection using plastic indexes［J］. IEEE Access， 2021，9：61955-61971.
［11］ TATA G， ROYER S J， POIRION O， et al. A robotic approach towards quantifying epipelagic bound plastic using deep visual models［J］. arXiv preprint arXiv：2105.01882， 2021.
［12］ HONG J S， FULTON M， SATTAR J. A generative approach towards improved robotic detection of marine litter［C］// 2020 IEEE International Conference on Robotics and Automation （ICRA）. IEEE， 2020：10525-10531.
［13］ PANWAR H， GUPTA P K， SIDDIQUI M K， et al. AquaVision： Automating the detection of waste in water bodies using deep transfer learning［J］. Case Studies in Chemical and Environmental Engineering， 2020，2. DOI： 10.1016/j.cscee.2020.100026.
［14］袁红春，臧天祺. 基于注意力机制及Ghost-YOLOv5的水下垃圾目标检测［J］. 环境工程， 2023，41（7）：214-221.
［15］ HIPOLITO J C， ALON A S， AMORADO R V， et al. Detection of underwater marine plastic debris using an augmented low sample size dataset for machine vision system： A deep transfer learning approach［C］// 2021 IEEE 19th Student Conference on Research and Development （SCOReD）. IEEE， 2021：82-86.
［16］ WU T H， WANG T W， LIU Y Q. Real-time vehicle and distance detection based on improved YOLOv5 network［C］// 2021 3rd World Symposium on Artificial Intelligence （WSAI）. IEEE， 2021：24-28.
［17］ YAN B， FAN P， LEI X Y， et al. A real-time apple targets detection method for picking robot based on improved YOLOv5［J］. Remote Sensing， 2021，13（9）. DOI： 10.3390/rs13091619.
［18］ BOCHKOVSKIY A， WANG C Y， LIAO H Y M. YOLOV4： Optimal speed and accuracy of object detection［J］. arXiv preprint arXiv：2004.10934， 2020.
［19］ ZHU X K， LYU S C， WANG X， et al. TPH-YOLOv5： Improved YOLOv5 based on transformer prediction head for object detection on drone-captured scenarios［C］// Proceedings of the 2021 IEEE/CVF International Conference on Computer Vision （ICCV）. IEEE， 2021：2778-2788.
［20］ WU C M， SUN Y Q， WANG T J， et al. Underwater trash detection algorithm based on improved YOLOv5s［J］. Journal of Real-Time Image Processing， 2022，19（5）：911-920.
［21］ STIMPER V， BAUER S， ERNSTORFER R， et al. Multidimensional contrast limited adaptive histogram equalization［J］. IEEE Access， 2019，7：165437-165447.
［22］ NGUYEN T P H， CAI Z， NGUYEN K， et al. Pre-processing image using brightening， CLAHE and RETINEX［J］. arXiv preprint arXiv：2003.10822， 2020.
［23］ YANG L X， ZHANG R Y， LI L D， et al. SimAM： A simple， parameter-free attention module for convolutional neural networks［C］// Proceedings of the 38th International Conference on Machine Learning. PMLR, 2021：11863-11874.
［24］ LI H L， LI J， WEI H B， et al. Slim-neck by GSConv： A better design paradigm of detector architectures for autonomous vehicles［J］. arXiv preprint arXiv：2206.02424， 2022.
［25］朱周华，齐琦. 基于改进YOLOv5s电动车头盔的自动检测与识别［J］. 计算机应用， 2023，43（4）：1291-1296.
［26］陈鑫林. 基于深度学习的水下垃圾检测［D］. 贵阳：贵州师范大学， 2022.
［27］王艺成，张国良，张自杰. 基于改进YOLOv5的小目标检测方法［J］. 计算机与现代化， 2023（5）：100-105.
［28］ GIRSHICK R， DONAHUE J， DARRELL T， et al. Rich feature hierarchies for accurate object detection and semantic segmentation［C］// 2014 IEEE Conference on Computer Vision and Pattern Recognition （CVPR）. IEEE， 2014：580-587.
［29］ LIU W， ANGUELOV D， ERHAN D， et al. SSD： Single shot multibox detector［C］// 2016 European Conference on Computer Vision （ECCV）. Springer， 2016：21-37.

[1]	ZHENG Jiuchao, ZHAO Xinyuan. Entity Linking Method Based on Topics and Description Information [J]. Computer and Modernization, 2024, 0(12): 10-14.
[2]	ZHAO Chenyang, XUE Tao, LIU Junhua. Fashion Clothing Pattern Generation Based on Improved Stable Diffusion [J]. Computer and Modernization, 2024, 0(12): 15-23.
[3]	WANG Xiaohang1, LI Yongjie1, YU Lei1, FAN Xiao2. A Method of Using Compound Event Probability Operation to Solve Problem of Negative Information Blocking Maximization [J]. Computer and Modernization, 2024, 0(12): 24-33.
[4]	ZHANG Simin, LIU Xinmei, YIN Junling, LI Baoling. PCB Defect Detection Method Based on Improved YOLOv7 [J]. Computer and Modernization, 2024, 0(12): 45-52.
[5]	ZHANG Xiaodong1, BAI Guangzhi1, LI Min1, LI Haoyang2. Oil and Gas Well Production Prediction Model Based on Empirical Wavelet Transform [J]. Computer and Modernization, 2024, 0(12): 53-58.
[6]	LIU Yunhai1, Feng Guang1, WU Xiaoting2, YANG Qun2 . Safety Helmet Wearing Detection Algorithm for Complex Construction Scenes [J]. Computer and Modernization, 2024, 0(12): 66-71.
[7]	LIU Baobao, YANG Jingjing, TAO Lu, WANG Heying . DSMSC Based on Attention Mechanism for Remote Sensing Image Scene Classification [J]. Computer and Modernization, 2024, 0(12): 72-77.
[8]	GU Yue, DENG Songfeng, SHEN Ji, MU Wentao, ZHAO Enqi. SAR Ship Detection Algorithm Based on Improved YOLOv8 [J]. Computer and Modernization, 2024, 0(12): 78-83.
[9]	WU Xiuling1, ZHOU Sheng1, WANG Chunjuan1, YU Cuizhuo2, LIU Hao3. Research Progress in Ultra Short-term Power Load Forecasting Technology [J]. Computer and Modernization, 2024, 0(12): 108-115.
[10]	LI Deyou1, 2, YU Jinsongdi1, 2, WEI Dandan1, 2, LUO Yuan1, 2, TONG Ruiju3. Abstract Tree Model for Gridded Cube Metadata [J]. Computer and Modernization, 2024, 0(11): 1-6.
[11]	GONG Yicheng1, 2, LIU Qing1, 2. Beijing Opera Binary Classification Based on RF-LCE-BiLSTM-Attention-AMSSA Model [J]. Computer and Modernization, 2024, 0(11): 7-12.
[12]	LI Taoying, LI Meng, WU Mengqiao. Taxi Passenger Flow Prediction Based on Heterogeneous Spatiotemporal Graph#br# Convolutional Networks [J]. Computer and Modernization, 2024, 0(11): 13-18.
[13]	ZHANG Tai1, YAN Zihao2, DUAN Jie2, ZHANG Zhihong2. Information Forwarding Strategy of Internet of Vehicles in Named Data Network [J]. Computer and Modernization, 2024, 0(11): 19-27.
[14]	YUAN Qingle, MU Li. Inventory Forecasting Method Based on Improved Elman Neural Network [J]. Computer and Modernization, 2024, 0(11): 28-33.
[15]	ZHANG Kun1, ZHANG Yongwei1, WU Yongcheng1, ZHANG Xiaowen2, ZHAI Shichen2. An LLM-based Method for Automatic Construction of Equipment Failure Knowledge Graphs [J]. Computer and Modernization, 2024, 0(11): 46-53.