A modified YOLOv4 detection method for a vision-based underwater garbage cleaning robot

2022, Volume 23, Issue 8

Abstract

Keywords

Related Research

Frontiers of Information Technology & Electronic Engineering >> 2022, Volume 23, Issue 8 doi: 10.1631/FITEE.2100473

A modified YOLOv4 detection method for a vision-based underwater garbage cleaning robot

Affiliation(s): First Research Institute of the Ministry of Public Security of PRC, Beijing 100048, China; School of Information Engineering, Minzu University of China, Beijing 100081, China; Department of Advanced Manufacturing and Robotics, College of Engineering, Peking University, Beijing 100871, China; State Key Laboratory of Management and Control for Complex Systems, Institute of Automation, Chinese Academy of Sciences, Beijing 100190, China; less

Received: 2021-10-01 Accepted: 2022-08-22 Available online: 2022-08-22

HTML69 PDF 24 Collect 0

Next Previous

Abstract

To tackle the problem of pollution, a vision-based autonomous underwater has been developed in our laboratory. We propose a garbage detection method based on a , allowing high-speed and high-precision . Specifically, the YOLOv4 algorithm is chosen as a basic neural network framework to perform . With the purpose of further improvement on the detection accuracy, YOLOv4 is transformed into a four-scale detection method. To improve the detection speed, model pruning is applied to the new model. By virtue of the improved detection methods, the robot can collect garbage autonomously. The detection speed is up to 66.67 frames/s with a mean average precision (mAP) of 95.099%, and experimental results demonstrate that both the detection speed and the accuracy of the improved YOLOv4 are excellent.

Keywords

Object detection ; Aquatic environment ; Garbage cleaning robot ; Modified YOLOv4

Related Research