This project uses the TrashBox dataset for training and validation.
Primary Source: TrashBox GitHub Repository
Author: Nikhil Venkat Kumsetty
Description: Dataset of trash objects for waste classification and detection, containing 17,785+ images across 7 categories.
Conference: 31st Conference of Open Innovations Association (FRUCT), University of Helsinki, Finland
Publication: "TrashBox: Trash Detection and Classification using Quantum Transfer Learning"
@inproceedings{kumsetty2022trashbox,
title={TrashBox: Trash Detection and Classification using Quantum Transfer Learning},
author={Kumsetty, Nikhil Venkat and others},
booktitle={31st Conference of Open Innovations Association (FRUCT)},
year={2022},
organization={University of Helsinki}
}Kumsetty, N. V., et al. (2022). TrashBox: Trash Detection and Classification using Quantum Transfer Learning.
In 31st Conference of Open Innovations Association (FRUCT). University of Helsinki, Finland.
Paper: MobileNetV2: Inverted Residuals and Linear Bottlenecks
Authors: Mark Sandler, Andrew Howard, Menglong Zhu, Andrey Zhmoginov, Liang-Chieh Chen
Published: CVPR 2018
@inproceedings{sandler2018mobilenetv2,
title={Mobilenetv2: Inverted residuals and linear bottlenecks},
author={Sandler, Mark and Howard, Andrew and Zhu, Menglong and Zhmoginov, Andrey and Chen, Liang-Chieh},
booktitle={Proceedings of the IEEE conference on computer vision and pattern recognition},
pages={4510--4520},
year={2018}
}Website: https://www.tensorflow.org/
Citation:
TensorFlow Developers. (2024). TensorFlow (v2.20) [Computer software]. https://doi.org/10.5281/zenodo.4724125
Website: https://flask.palletsprojects.com/
Citation:
Pallets Projects. (2024). Flask (v3.1.2) [Computer software]. https://palletsprojects.com/p/flask/
- Pan, S. J., & Yang, Q. (2010). A survey on transfer learning. IEEE Transactions on knowledge and data engineering, 22(10), 1345-1359.
- Kumsetty, N. V., et al. (2022). TrashBox: Trash Detection and Classification using Quantum Transfer Learning.
- Aral, R. A., et al. (2018). Classification of TrashNet dataset based on deep learning models.
If you use or reference this TrashFormer implementation, please cite:
@software{trashformer2025,
title={TrashFormer: Smart Waste Classification System},
author={Your Name},
year={2025},
url={https://github.com/yourusername/trashformer},
note={AI-powered waste classification using MobileNetV2 transfer learning}
}TrashFormer is released under the MIT License.
The TrashBox dataset is used under its original license terms. Please refer to the TrashBox repository for dataset licensing information.
Special thanks to:
- Nikhil Venkat Kumsetty - For creating and sharing the TrashBox dataset
- IEEE FRUCT Conference - For publishing waste classification research
- Google Brain Team - For TensorFlow and MobileNetV2
- Open Source Community - For tools and frameworks used in this project
Last Updated: October 12, 2025