EWasteNet: A Two-Stream Data Efficient Image Transformer Approach for E-Waste Classification
arXiv:2311.12823 · doi:10.1109/ICSECS58457.2023.10256323
Abstract
Improper disposal of e-waste poses global environmental and health risks, raising serious concerns. The accurate classification of e-waste images is critical for efficient management and recycling. In this paper, we have presented a comprehensive dataset comprised of eight different classes of images of electronic devices named the E-Waste Vision Dataset. We have also presented EWasteNet, a novel two-stream approach for precise e-waste image classification based on a data-efficient image transformer (DeiT). The first stream of EWasteNet passes through a sobel operator that detects the edges while the second stream is directed through an Atrous Spatial Pyramid Pooling and attention block where multi-scale contextual information is captured. We train both of the streams simultaneously and their features are merged at the decision level. The DeiT is used as the backbone of both streams. Extensive analysis of the e-waste dataset indicates the usefulness of our method, providing 96% accuracy in e-waste classification. The proposed approach demonstrates significant usefulness in addressing the global concern of e-waste management. It facilitates efficient waste management and recycling by accurately classifying e-waste images, reducing health and safety hazards associated with improper disposal.
6 pages
References in corpus (7)
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
- Predicting multicellular function through multi-layer tissue networks
- Simple and Deep Graph Convolutional Networks
- Everything is Connected: Graph Neural Networks
- Vision GNN: An Image is Worth Graph of Nodes
- A Survey on Oversmoothing in Graph Neural Networks
- Distinguishing Human Generated Text From ChatGPT Generated Text Using Machine Learning