Skip to content

khalld/trashbin-classifier

Repository files navigation

Trash bin classifier

Model that classify watebin in full, empty, half.

Dataset is available here send a message before using it.

Setup

Create virtual environment

python3 -m venv venv

Activate venv

source venv/bin/activate

Install requirements libs

pip install -r requirements.txt

Run app

python3 app.py

Utils

Exit from venv

deactivate

Tensorboard logs:

tensorboard --logdir logs

About

Trash bin classifier using PyTorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •