Muliti_Lable_Classification_Planet

Try to use multi-label-classification realize Planet-Kaggle task

Dataset

• https://www.kaggle.com/c/planet-understanding-the-amazon-from-space/data
• https://www.kaggle.com/nikitarom/planets-dataset

Config

vim ./configs/cfgs.py

EDA

Planet Dataset EDA

Preprocess

step1: Count dataset

python ./tools/preprocess.py

step2: Get mean and std

python ./tools/get_mean_std.py

Train

python ./train.py

• training

* epoch

* learning rate

Inference

python ./inference.py

Bag of Tricks

• Multi-fold cross-validation

• Re-Weighting strategy to alleviate the class-imbalance

• Get best threshold with brute force search

• Stop when loss plateaus

• Image Augmentation

• ...

Deployment

Deployment-with-tensorRT

TODO

Reference

• https://github.com/rwightman/pytorch-planet-amazon