Waste Detection System Based on Data Augmentation and YOLO_EC

Abstract

The problem of waste classification has been a major concern for both the government and society, and whether waste can be effectively classified will affect the sustainable development of human society. To perform fast and efficient detection of waste targets in the sorting process, this paper proposes a data augmentation + YOLO_EC waste detection system. First of all, because of the current shortage of multi-objective waste classification datasets, the heavy workload of human data collection, and the limited improvement of data features by traditional data augmentation methods, DCGAN (deep convolution generative adversarial networks) was optimized by improving the loss function, and an image-generation model was established to realize the generation of multi-objective waste images; secondly, with YOLOv4 (You Only Look Once version 4) as the basic model, EfficientNet is used as the backbone feature extraction network to realize the light weight of the algorithm, and at the same time, the CA (coordinate attention) attention mechanism is introduced to reconstruct the MBConv module to filter out high-quality information and enhance the feature extraction ability of the model. Experimental results show that on the HPU_WASTE dataset, the proposed model outperforms other models in both data augmentation and waste detection.

Enhancing Mobile Acute Lymphoblastic Cancer Detection: Transfer Learning from YOLOv9 to TensorFlow for Real-Time Applications

“…Table 1 details the specific augmentation operations used. The application of these data augmentation techniques resulted in the expansion of the dataset with a multitude of diverse variations derived from the original images, thereby creating a larger and more diversified training set [6].…”

YOLOv5-KCB: A New Method for Individual Pig Detection Using Optimized K-Means, CA Attention Mechanism and a Bi-Directional Feature Pyramid Network

“…In this article, we will use common evaluation metrics in deep learning [43,44], including recall (R), precision (P), average precision (AP), and mAP, which is the average of AP values across all categories. TP (true positives) is the number of correctly identified individual pigs, while FN (false negatives) is the number of missed individual pigs, and FP (false positives) is the number of falsely identified individual pigs.…”

A Systematic Literature Review of Waste Identification in Automatic Separation Systems

“…Ref. [100] proposes using deep convolution generative adversarial networks (DCGANs) to generate synthetic samples with real ones to train a YoloV4 detector model. As a result, using data augmentation improves the mAP by 4.54% compared to using only real samples.…”