fast-reid/datasets/README.md

44 lines
1.4 KiB
Markdown
Raw Normal View History

2020-05-12 23:00:15 +08:00
# Setup Buildin Dataset
Fastreid has buildin support for a few datasets. The datasets are assumed to exist in a directory specified by the environment variable `FASTREID_DATASETS`. Under this directory, fastreid expects to find datasets in the structure described below.
You can set the location for builtin datasets by `export FASTREID_DATASETS=/path/to/datasets/`. If left unset, the default is `datasets/` relative to your current working directory.
2020-05-12 23:00:15 +08:00
The [model zoo](https://github.com/JDAI-CV/fast-reid/blob/master/MODEL_ZOO.md) contains configs and models that use these buildin datasets.
2020-05-12 23:00:15 +08:00
## Expected dataset structure for Market1501
1. Download dataset to `datasets/` from [baidu pan](https://pan.baidu.com/s/1ntIi2Op) or [google driver](https://drive.google.com/file/d/0B8-rUzbwVRk0c054eEozWG9COHM/view)
2. Extract dataset. The dataset structure would like:
```bash
datasets/
Market-1501-v15.09.15/
bounding_box_test/
bounding_box_train/
```
## Expected dataset structure for DukeMTMC
1. Download datasets to `datasets/`
2. Extract dataset. The dataset structure would like:
```bash
datasets/
DukeMTMC-reID/
bounding_box_train/
bounding_box_test/
```
2020-05-12 23:00:15 +08:00
## Expected dataset structure for MSMT17
1. Download datasets to `datasets/`
2. Extract dataset. The dataset structure would like:
```bash
datasets/
MSMT17_V2/
mask_train_v2/
mask_test_v2/
```