History

Xingyu Liao 664ba4ae11 save idx2class dict when training classification (#444 )		2021-03-26 20:51:52 +08:00
..
configs	save idx2class dict when training classification (#444 )	2021-03-26 20:51:52 +08:00
fastcls	save idx2class dict when training classification (#444 )	2021-03-26 20:51:52 +08:00
README.md	support classification in fastreid (#443 )	2021-03-26 20:17:39 +08:00
train_net.py	save idx2class dict when training classification (#444 )	2021-03-26 20:51:52 +08:00

README.md

FastCls in FastReID

This project provides a baseline and example for image classification based on fastreid.

Datasets Preparation

We refer to pytorch tutorial for dataset preparation. This is just an example for building a classification task based on fastreid. You can customize your own datasets and model.

Usage

If you want to train models with 4 gpus, you can run

python3 projects/FastCls/train_net.py --config-file projects/FastCls/config/base-cls.yml --num-gpus 4