https://github.com/deepdiy/trash-type-detection-software

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
https://github.com/deepdiy/trash-type-detection-software

Last synced: 3 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/deepdiy/trash-type-detection-software
• Owner: deepdiy
• Created: 2019-05-20T04:15:44.000Z (about 7 years ago)
• Default Branch: master
• Last Pushed: 2020-10-01T05:18:16.000Z (over 5 years ago)
• Last Synced: 2025-06-14T11:43:12.675Z (12 months ago)
• Language: Jupyter Notebook
• Size: 32.2 MB
• Stars: 6
• Watchers: 1
• Forks: 1
• Open Issues: 2
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Trash Type Detecion Software

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.

![demo](https://i.imgur.com/wFttTnb.png)

![demo](https://i.imgur.com/QD2C5TC.jpg)

![demo2](https://i.imgur.com/HYl1Z2c.jpg)

# Installation

```python

pip install -r requirements.txt

```

# Run App

```python

python app/main.py

```

# Capture image without detection

1. Run App

2. Click `Capture` button in window

# Training model

[Follow this instruction](https://github.com/deepdiy/trash-type-detection-software/tree/master/tf_ssd_mobilenet)

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

# Use new model

put `frozen_inference_graph.pb` in `/app/model`