If you want to train the models locally install all the necessary packages with pip.
pip3 install -r requirements.txt
You can find the associated Images in the first release v0.1. Download the .zip and put your repisitory in the following folder structure:
βββ TrashTron2.ipynb
βββ datasets
| βββ train/ # training images
| βββ val/ # validation iamges
| βββ train.json # training annotations
| βββ val.json # validation annotations
βββ output
βββ coco_instances_results.json
βββ events.out.tfevents
βββ instances_predictions.pth
βββ last_checkpoint
βββ metrics.json
βββ model_final.pth # final model after training
You can use the whole project in the associated TrashTron2 Notebook. You can apply the folder structure from above.