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).

近期下载者

相关文件


收藏者