classifier_training

small tool to annotate data and train xgboost classifiers

How to use

1. Get a CSV with two columns (raw and mask) of your training data
2. Change the paths in annotate/tool.ipynb, change the classes, etc ...
3. Run the notebook
4. Run the last cell independently whenever you want to save your labels
5. Finally, in train_classifier.ipynb change the classes and all the things you would like to change and run.