Close

Model training for of garbage in forest

A project log for AkerfaBot

A DIY robot that helps collect the trash in the woods. Image recognition will be uses to detect trash and collect it using a robotic arm.

nait-malek-youssefNait Malek Youssef 07/13/2022 at 22:100 Comments

The first step in this project was to train a model to detect garbage in the forest.  To gain some time, I tried to see if there is a prepared dataset for garbage detection in forest, and I found this github repo Garbage Detection using Yolov5 on Jetson Nano. This model tries to detect various types of garbage objects, which is suitable for this project. However, when cloned and tested using new images, the trained model can detect some objects correctly and give them the correct annotation. But sometimes, the model fails to detect objects or detects wrong objects.

For test purposes, I will use this model for the next build steps of the robot.  And when done, I will prepare a clean dataset for garbage detection in the forest and beach.


Discussions