My code for Cdiscount's Image Classification Challenge. Tested on a subset of 10k samples from the ~7m on the full dataset.
- Keras 2.0 w/ TF backend
- sklearn
- skimage
- tqdm
- h5py
- imgaug
- In
params.pysetbase_dirto your working directory. You can also set the model to use and training parameters. - Place train.bson and test.bson in {work_dir}/input.
- Run
read_data_train.pyandread_data_test.pyto read and unpack train data and test data respectively. - Run
train.pyto train the model andtest_submit.pyto predict on test data and generate submission file.