2021-04-08 11:01:15 +08:00
|
|
|
# FastClas in FastReID
|
2021-01-18 11:36:38 +08:00
|
|
|
|
|
|
|
|
This project provides a baseline and example for image classification based on fastreid.
|
|
|
|
|
|
|
|
|
|
## Datasets Preparation
|
|
|
|
|
|
|
|
|
|
We refer to [pytorch tutorial](https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html) 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
|
|
|
|
|
```bash
|
2021-03-30 15:47:14 +08:00
|
|
|
python3 projects/FastClas/train_net.py --config-file projects/FastClas/config/base-clas.yml --num-gpus 4
|
2021-01-18 11:36:38 +08:00
|
|
|
```
|
