This project was developed for the CNN workshop of the Robotic Vision Summer School 2017 (see http://roboticvision.org/events/rvss-summer-school/). This workshop taught how to train a CNN classifier for user-defined classes using keras and tensorflow (See the presentation).
- Python 3
- Tensorflow 0.12
- keras 1.2.2
Create a data directory with train and validation folder. In each folder, add subfolders for each class and name it with the class label. Then, pass the data directory to the train and validation script. Note that you can always train with subsets of classes by passing -classes option.
Train and Valuation
Script to run train and validation. For instructions,
$ python ./src/objcls_train_val.py --help
Script to peform prediction on test data. For instructions,
$ python ./src/objcls_predict.py --help