Classify trash by computer vision. A ssd-mobilenet based deep learning model and a kivy based GUI is included in this repository. Currently, the software can detect bottles in video frame.
pip install -r requirements.txtpython app/main.py- Run App
- Click
Capturebutton in window
The notebook for training is here, you can open the notebook in Google Colab: https://github.com/deepdiy/trash-type-detection-software/blob/master/tf_ssd_mobilenet/tensorflow_object_detection_training_colab_trash_detection.ipynb
put frozen_inference_graph.pb in /app/model