Clothes Classification

Instalation

• install requirements
• install mmclassification: link

Train model:

python train.py

Evaluation

• By default, mmclassification provides evaluation metrices during training. To see run:

python display_stats.py

you should see:

Visualization:

• Run:

python model_inference.py

Notes

• To activate transfer learning specify in train_config.py:

TRANSFER_LEARNING = True

• pth file for experiment "clothes_experiment" is stored here
• pth file for experiment "clothes_experiment_tf" is stored here
• transfer learning checkpoint download link, after download place to src/checkopints
• images dataset link