Improved YOLO Based Detection Algorithm for Floating Debris in Waterway

• Main Authors: Feng Lin, Tian Hou, Qiannan Jin, Aiju You
• Format: Article
• Language: English
• Published: MDPI AG 2021-08-01
• Series: Entropy
• Subjects: deep learning; YOLOv5; floating debris detection; detection algorithm
• Online Access: https://www.mdpi.com/1099-4300/23/9/1111

Various floating debris in the waterway can be used as one kind of visual index to measure the water quality. The traditional image processing method is difficult to meet the requirements of real-time monitoring of floating debris in the waterway due to the complexity of the environment, such as reflection of sunlight, obstacles of water plants, a large difference between the near and far target scale, and so on. To address these issues, an improved YOLOv5s (FMA-YOLOv5s) algorithm by adding a feature map attention (FMA) layer at the end of the backbone is proposed. The mosaic data augmentation is applied to enhance the detection effect of small targets in training. A data expansion method is introduced to expand the training dataset from 1920 to 4800, which fuses the labeled target objects extracted from the original training dataset and the background images of the clean river surface in the actual scene. The comparisons of accuracy and rapidity of six models of this algorithm are completed. The experiment proves that it meets the standards of real-time object detection.