DukeMTMC-VideoReID-master
所属分类:其他
开发工具:Python
文件大小:37KB
下载次数:11
上传日期:2018-10-27 21:58:36
上 传 者:
gohard
说明: dukeMTMC视频数据集行人再识别代码,基于python
(DukeMTMC video dataset person re-identification code, based on python)
文件列表:
baseline.py (5354, 2018-05-07)
LICENSES (0, 2018-05-07)
LICENSES\LICENSE_DukeMTMC-VideoReID.txt (2003, 2018-05-07)
LICENSES\LICENSE_DukeMTMC.txt (1602, 2018-05-07)
LICENSES\LICENSE_ETAP.txt (1062, 2018-05-07)
logs (0, 2018-05-07)
reid (0, 2018-05-07)
reid\datasets (0, 2018-05-07)
reid\datasets\dukemtmc_videoReID.py (3482, 2018-05-07)
reid\datasets\mars.py (3670, 2018-05-07)
reid\datasets\__init__.py (1210, 2018-05-07)
reid\dist_metric.py (926, 2018-05-07)
reid\evaluation_metrics (0, 2018-05-07)
reid\evaluation_metrics\classification.py (521, 2018-05-07)
reid\evaluation_metrics\ranking.py (4735, 2018-05-07)
reid\evaluation_metrics\__init__.py (168, 2018-05-07)
reid\evaluators.py (4240, 2018-05-07)
reid\feature_extraction (0, 2018-05-07)
reid\feature_extraction\cnn.py (751, 2018-05-07)
reid\feature_extraction\database.py (1311, 2018-05-07)
reid\feature_extraction\__init__.py (180, 2018-05-07)
reid\loss (0, 2018-05-07)
reid\loss\oim.py (1727, 2018-05-07)
reid\loss\triplet.py (1344, 2018-05-07)
reid\loss\tri_clu_loss.py (1682, 2018-05-07)
reid\loss\__init__.py (255, 2018-05-07)
reid\metric_learning (0, 2018-05-07)
reid\metric_learning\euclidean.py (425, 2018-05-07)
reid\metric_learning\kissme.py (1654, 2018-05-07)
reid\metric_learning\__init__.py (653, 2018-05-07)
reid\models (0, 2018-05-07)
reid\models\end2end.py (2385, 2018-05-07)
reid\models\resnet.py (4439, 2018-05-07)
reid\models\__init__.py (1666, 2018-05-07)
reid\trainers.py (3390, 2018-05-07)
reid\utils (0, 2018-05-07)
reid\utils\data (0, 2018-05-07)
... ...
# DukeMTMC-VideoReID
DukeMTMC-VideoReID [1] is a subset of the [DukeMTMC](http://vision.cs.duke.edu/DukeMTMC/) tracking dataset [2] for video-based person re-identification.
The dataset consists of 702 identities for training, 702 identities for testing, and 408 identities as distractors. In total there are 2,196 videos for training and 2,636 videos for testing. Each video contains person images sampled every 12 frames. During testing, a video for each ID is used as the query and the remaining videos are placed in the gallery.
### Download Dataset
You can download the DukeMTMC-VideoReID dataset from
[[Direct Link]](http://vision.cs.duke.edu/DukeMTMC/data/misc/DukeMTMC-VideoReID.zip) [[Google Drive]](https://drive.google.com/open?id=1Fdu5GK-C7P8M9QiLbiQNyT_RUFt8oFco) [[BaiduYun]](https://pan.baidu.com/s/1qL39rnjTjyzjqaD-Wuv8KQ).
### About Dataset
|Directory | Description |
| -------- | ----- |
|./train | The training video tracklets. It contains 702 identities.|
|./query | The query video tracklets. Each of them is from different identities in different cameras.|
|./gallery | The gallery video tracklets. It contains 702 gallery identities and 408 distractors.|
### Directory Structure
Followings are the directory structure for DukeMTMC-VideoReID.
> Splits
>> Person ids
>>> Video tracklet ids
>>>> Frame bounding box images
For example, for one frame image `train/0001/0003/0001_C6_F0099_X30823.jpg`, `train`, `0001`, `0003`, and `0001_C6_F0099_X30823.jpg` are the split, person id, video tracklet id, and image frame name, respectively.
**Naming Rules for image file.**
For most frame bounding box images, e.g. `0001_C6_F0099_X30823.jpg`, "0001" is the identity. "C6" indicate Camera 6. "F0099" means it is the 99th frame within the tracklet. "X" indicates it is a normal image (otherwise, "D" for distractors) and " 30823" is the 30823th frame in the whole video of Camera 6.
## Training Baseline model
### ETAP-Net
The baseline model is an end-to-end ResNet-50 model with temporal average pooling (ETAP-Net).
More details about the ETAP-Net can be found in our CVPR2018 paper [Exploit the Unknown Gradually: One-Shot Video-Based Person Re-Identification by Stepwise Learning](https://yu-wu.net/pdf/CVPR2018_Exploit-Unknown-Gradually.pdf).
### Dependencies
- Python 3.6
- PyTorch (version >= 0.2.0)
- h5py, scikit-learn, metric-learn, tqdm
### Run
Move the downloaded dataset file `DukeMTMC-VideoReID.zip` to `./data/` and unzip here.
```shell
python3 run.py --dataset DukeMTMC-VideoReID --logs_dir logs/DukeMTMC-VideoReID_baseline/ --max_frames 900 --batch_size 16
```
### Results on DukeMTMC-VideoReID
Method |
Rank-1 |
Rank-5 |
Rank-20 |
mAP |
ETAP-Net |
83.62 |
94.59 |
97.58 |
78.34 |
### References
- [1] Exploit the Unknown Gradually: One-Shot Video-Based Person Re-Identification by Stepwise Learning. Wu et al., CVPR 2018
- [2] Performance Measures and a Data Set for Multi-Target, Multi-Camera Tracking. Ristani et al., ECCVWS 2016
Please cite the following two papers if this dataset helps your research.
```
@inproceedings{wu2018cvpr_oneshot,
title = {Exploit the Unknown Gradually: One-Shot Video-Based Person Re-Identification by Stepwise Learning},
author = {Wu, Yu and Lin, Yutian and Dong, Xuanyi and Yan, Yan and Ouyang, Wanli and Yang, Yi},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2018}
}
@inproceedings{ristani2016MTMC,
title = {Performance Measures and a Data Set for Multi-Target, Multi-Camera Tracking},
author = {Ristani, Ergys and Solera, Francesco and Zou, Roger and Cucchiara, Rita and Tomasi, Carlo},
booktitle = {European Conference on Computer Vision workshop on Benchmarking Multi-Target Tracking},
year = {2016}
}
```
### License
Please refer to the license file for [DukeMTMC-VideoReID](https://github.com/Yu-Wu/DukeMTMC-VideoReID/blob/master/LICENSES/LICENSE_DukeMTMC-VideoReID.txt) and [DukeMTMC](https://github.com/Yu-Wu/DukeMTMC-VideoReID/blob/master/LICENSES/LICENSE_DukeMTMC.txt).
近期下载者:
相关文件:
收藏者: