Research on Data Augmentation for Garbage Classification with Target Detection Algorithm

Abstract

Garbage classification is an essential work in daily life. With the development of artificial intelligence (AI), we have begun to use object detection to achieve garbage classification. However, the stacking and occlusion of garbage often make the detection performance unsatisfactory. In this paper, YOLOv5 algorithm is used for garbage detection and classification. YOLOv5 has three detection heads, which can detect objects at multiple scales. At the same time, Mosaic and Coarse Dropout image augmentation techniques are used to process our private data set of garbage, which improves the diversity of images. In this way, the target classification accuracy of the detection model can reach 75.7%, which is significantly improved compared with 69.9% without data augmentation. This paper provides a new technical support for the scene classification of stacked garbage based on deep learning.