History

Xingyu Liao fb36b23678 bugfix for attribute project (#450 ) Summary: refactor sample weight in attribute recognition; change all options to False in defaults.py and modify yaml files		2021-03-31 17:07:19 +08:00
..
configs	bugfix for attribute project (#450 )	2021-03-31 17:07:19 +08:00
fastclas	bugfix for attribute project (#450 )	2021-03-31 17:07:19 +08:00
README.md	bugfix & merge classification transforms (#448 )	2021-03-30 15:47:14 +08:00
train_net.py	bugfix for attribute project (#450 )	2021-03-31 17:07:19 +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/FastClas/train_net.py --config-file projects/FastClas/config/base-clas.yml --num-gpus 4