hypergraph_reid
Implementation of "Learning Multi-Granular Hypergraphs for Video-Based Person Re-Identification" If you find this help your research, please cite
@inproceedings{DBLP:conf/cvpr/YanQC0ZT020,
author = {Yichao Yan and
Jie Qin and
Jiaxin Chen and
Li Liu and
Fan Zhu and
Ying Tai and
Ling Shao},
title = {Learning Multi-Granular Hypergraphs for Video-Based Person Re-Identification},
booktitle = {2020 {IEEE/CVF} Conference on Computer Vision and Pattern Recognition,
{CVPR} 2020, Seattle, WA, USA, June 13-19, 2020},
pages = {2896--2905},
publisher = {{IEEE}},
year = {2020}
}
Installation
We use python 3.7 and pytorch=0.4
Data preparation
All experiments are done on MARS, as it is the largest dataset available to date for video-based person reID. Please follow deep-person-reid to prepare the data. The instructions are copied here:
- Create a directory named
mars/
underdata/
. - Download dataset to
data/mars/
from http://www.liangzheng.com.cn/Project/project_mars.html. - Extract
bbox_train.zip
andbbox_test.zip
. - Download split information from https://github.com/liangzheng06/MARS-evaluation/tree/master/info and put
info/
indata/mars
(we want to follow the standard split in [8]). The data structure would look like:
mars/
bbox_test/
bbox_train/
info/
Usage
To train the model, please run
sh run_hypergraphsage_part.sh
Performance
Trained model [Google]
The shared trained model achieves 85.6% mAP and 89.5% rank-1 accuracy. According to my training log, the best model achieves 86.2% mAP and 90.0% top-1 accuracy. This may need adjustion in hyperparameters.
Acknowledgements
Our code is developed based on Video-Person-ReID (https://github.com/jiyanggao/Video-Person-ReID).