mirror of
https://github.com/JDAI-CV/fast-reid.git
synced 2025-06-03 14:50:47 +08:00
76 lines
2.9 KiB
Markdown
76 lines
2.9 KiB
Markdown
# DSR in FastReID
|
||
**Deep Spatial Feature Reconstruction for Partial Person Re-identification**
|
||
|
||
Lingxiao He, Xingyu Liao
|
||
|
||
[[`CVPR2018`](http://openaccess.thecvf.com/content_cvpr_2018/papers/He_Deep_Spatial_Feature_CVPR_2018_paper.pdf)] [[`BibTeX`](#CitingDSR)]
|
||
|
||
**Foreground-aware Pyramid Reconstruction for Alignment-free Occluded Person Re-identification**
|
||
|
||
Lingxiao He, Xingyu Liao
|
||
|
||
[[`ICCV2019`](http://openaccess.thecvf.com/content_ICCV_2019/papers/He_Foreground-Aware_Pyramid_Reconstruction_for_Alignment-Free_Occluded_Person_Re-Identification_ICCV_2019_paper.pdf)] [[`BibTeX`](#CitingFPR)]
|
||
|
||
## News！
|
||
|
||
[1] The old_version code can be check in [old_version](https://github.com/JDAI-CV/Partial-Person-ReID), you can obtain the same result published in paper, and the new version code is updating, please waiting!
|
||
|
||
## Installation
|
||
|
||
First install FastReID, and then put Partial Datasets in directory datasets. The whole framework of FastReID-DSR is
|
||
<div align="center">
|
||
<img src="https://firebasestorage.googleapis.com/v0/b/firescript-577a2.appspot.com/o/imgs%2Fapp%2FSherlockWorkspace%2F1nVTE3Sn5c.jpg?alt=media&token=e7e9fcfc-4fc1-49c8-bcf4-c007028fdd25" width="700px" />
|
||
</div>
|
||
|
||
and the detail you can refer to
|
||
## Datasets
|
||
|
||
The datasets can find in [Google Drive](https://drive.google.com/file/d/1p7Jvo-RJhU_B6hf9eAhIEFNhvrzM5cdh/view?usp=sharing)
|
||
|
||
PartialREID---gallery: 300 images of 60 ids, query: 300 images of 60 ids
|
||
|
||
PartialiLIDS---gallery: 119 images of 119 ids, query: 119 images of 119 ids
|
||
|
||
OccludedREID---gallery: 1,000 images of 200 ids, query: 1,000 images of 200 ids
|
||
|
||
## Training and Evaluation
|
||
|
||
To train a model, run:
|
||
```bash
|
||
python3 projects/PartialReID/train_net.py --config-file <config.yaml>
|
||
```
|
||
|
||
For example, to train the re-id network with IBN-ResNet-50 Backbone
|
||
one should execute:
|
||
```bash
|
||
CUDA_VISIBLE_DEVICES='0,1,2,3' python3 projects/PartialReID/train_net.py --config-file 'projects/PartialReID/configs/partial_market.yml'
|
||
```
|
||
|
||
## Results
|
||
|
||
| Method | PartialREID | OccludedREID | PartialiLIDS |
|
||
|:--:|:--:|:--:|:--:|
|
||
| | Rank@1 (mAP)| Rank@1 (mAP)| Rank@1 (mAP)|
|
||
| DSR (CVPR’18) |73.7(68.1) |72.8(62.8)|64.3(58.1)|
|
||
| FPR (ICCV'19) | 81.0(76.6)|78.3(68.0)|68.1(61.8)|
|
||
| FastReID-DSR | 82.7(76.8)|81.6(70.9)|73.1(79.8) |
|
||
|
||
## <a name="CitingDSR"></a >Citing DSR and Citing FPR
|
||
|
||
If you use DSR or FPR, please use the following BibTeX entry.
|
||
|
||
```
|
||
@inproceedings{he2018deep,
|
||
title={Deep spatial feature reconstruction for partial person re-identification: Alignment-free approach},
|
||
author={He, Lingxiao and Liang, Jian and Li, Haiqing and Sun, Zhenan},
|
||
booktitle={IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
|
||
year={2018}
|
||
}
|
||
@inproceedings{he2019foreground,
|
||
title={Foreground-aware Pyramid Reconstruction for Alignment-free Occluded Person Re-identification},
|
||
author={He, Lingxiao and Wang, Yinggang and Liu, Wu and Zhao, He and Sun, Zhenan and Feng, Jiashi},
|
||
booktitle={IEEE International Conference on Computer Vision (ICCV)},
|
||
year={2019}
|
||
}
|
||
```
|