All codes for cdiscount image classification.
Data visualization and analysis: site
python main.py --train_bson_path <the absolute path of training data>
| file name | notes |
|---|---|
| data transform | contain various data augmentation functions |
| label id dict | two dictions, mapping from category id to category label (a integer between 1 and 5270) and from category label to category id respectively |
| se inception v3 | network |
| Trainer | The first trainer ensembling all parts together to train the model, but suffers from poor flexibility and need to be improved |
| train se inv3 | The improved trainer, but still exist some bugs |
| utils | contain important useful functions, including data loading and learning rate adjustment |