Garbage Classification and Detection Based on Improved YOLOv7 Network

Abstract

With the improvement of people’s living standards, garbage classification is gradually forced. However, due to people’s awareness and knowledge, the classification accuracy and disposal of garbage are difficult to keep pace with guideline changes. With the consideration of the problems of low efficiency, heavy task and poor environment of garbage manual classification, an improved YOLOv7 target detection method is proposed to realize the effective classification of garbage. In this study, the recursive gated convolutional gnconv was used to establish the HorNet network architecture, and the model was trained by making specific data sets. The C3HB module is added to the YOLO model, and the pooling layer is optimized to replace SPPFCSPC to improve the detection accuracy of the target. The experimental results show that the garbage detection and classification method proposed in this study has excellent accuracy. Experiments show that the map value, accuracy and recall rate of the proposed model on garbage datasets are 99.25%, 99.33% and 98.03%, respectively, which are 1.50%, 3.99% and 1.41% higher than those of YOLOv7. The overall results are better than the original model.