Skip to content

Latest commit

Β 

History

History
30 lines (23 loc) Β· 1.13 KB

GettingStarted.md

File metadata and controls

30 lines (23 loc) Β· 1.13 KB

Getting started

Requirements

If you want to train the models locally install all the necessary packages with pip.

pip3 install -r requirements.txt

Weights & Images

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

Google Collab

You can use the whole project in the associated TrashTron2 Notebook. You can apply the folder structure from above.