This project aims to develop an intelligent waste segregation system that can automatically classify and sort different types of waste materials into three categories: organic, inorganic, and metal. The system utilizes computer vision and machine learning techniques to analyze captured images of waste items and direct them into the appropriate bin section using a mechanical sorting mechanism.
Improper waste management and disposal have become critical environmental issues. This project addresses these challenges by automating the waste segregation process, reducing the burden on manual labor and improving the overall efficiency of waste management practices.
The proposed solution involves the integration of several key components: image acquisition using a Raspberry Pi camera module, object detection and classification using the YOLO (You Only Look Once) model, Raspberry Pi integration for inference and control, a mechanical sorting mechanism with stepper motors and flaps, and a user-friendly web interface.
- Raspberry Pi 3 Model B+
- Raspberry Pi Camera Module
- Nemo 17 Stepper Motors and A4988 Motor Drivers
- Mechanical Sorting Mechanism (Flaps and Bins)
- Printed Circuit Boards (PCBs) for motor drivers and control circuits
- Power supply units
- Enclosure materials (e.g., acrylic, metal sheets)
-
Install the required libraries and dependencies on the Raspberry Pi:
- Python (usually pre-installed on Raspbian)
- Ultralytics YOLO library:
pip install ultralytics
- Other required libraries (e.g., numpy, opencv-python)
-
Transfer the trained YOLO model (
sbin_best.pt
) to the Raspberry Pi. -
Set up the mechanical sorting mechanism and connect the stepper motors to the appropriate GPIO pins.
-
Clone the project repository and navigate to the project directory.
- Run the main script (
motors_come_back.py
) to start the waste segregation system. - The Raspberry Pi camera will capture images of waste items as they are introduced into the system.
- The captured images will be processed by the YOLO model to classify the waste into organic, inorganic, or metal categories.
- Based on the predicted category, the mechanical sorting mechanism will be activated to direct the waste item into the corresponding bin section.
- The real-time classification results and insights will be displayed on the web interface.
- The intelligent waste segregation system has been successfully developed and integrated with the Raspberry Pi and mechanical sorting mechanism.
- The trained YOLO model achieves an accuracy of [X%] in classifying waste items into the three categories.
- The system can process and sort [Y] waste items per minute.
- [https://drive.google.com/file/d/14Xc3c1qOiZjQKldAli8NWx0QTQpTw1aF/view?usp=sharing]
If you'd like to contribute to this project, please follow these steps:
- Fork the repository.
- Create a new branch (
git checkout -b feature/your-feature
). - Make your changes and commit them (
git commit -am 'Add your feature'
). - Push to the branch (
git push origin feature/your-feature
). - Create a new Pull Request.