Competition: image classification with 128 classes Link: https://www.kaggle.com/c/imaterialist-challenge-furniture-2018
- Download data from kaggle to
./data/
- Train model
python cnn_runner_pth.py train model_name augmentation_mixup mixup_alpha
- Predict
python cnn_runner_pth.py predict model_name augmentation_mixup mixup_alpha
- Calculate weights for finetuned models
python statistics.py
- Generate submission
python submit.py
- ensemble denset201, inceptionv4, inceptionresnetv2, senet154, se_resnext101, naslarge
- hflip at test
- Public Borard: 9/436 Private Board: 13/436
- models trained at model
- skrypka : winner of imaterialist furniture 2018