fast-reid/datasets/README.md

44 lines
1.7 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
2021-02-10 15:29:09 +08:00
## Expected dataset structure for [Market1501](https://www.cv-foundation.org/openaccess/content_iccv_2015/papers/Zheng_Scalable_Person_Re-Identification_ICCV_2015_paper.pdf)
2020-05-12 23:00:15 +08:00
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/
```
2021-02-10 15:29:09 +08:00
## Expected dataset structure for [DukeMTMC-reID](https://openaccess.thecvf.com/content_ICCV_2017/papers/Zheng_Unlabeled_Samples_Generated_ICCV_2017_paper.pdf)
2020-05-12 23:00:15 +08:00
1. Download datasets to `datasets/`
2. Extract dataset. The dataset structure would like:
```bash
datasets/
DukeMTMC-reID/
bounding_box_train/
bounding_box_test/
```
2021-02-10 15:29:09 +08:00
## Expected dataset structure for [MSMT17](https://arxiv.org/abs/1711.08565)
1. Download datasets to `datasets/`
2. Extract dataset. The dataset structure would like:
```bash
datasets/
MSMT17_V2/
mask_train_v2/
mask_test_v2/
```