Create Readme.md

pull/66/head
xbq1994 2020-05-27 22:27:33 +08:00 committed by GitHub
parent 41f7bb772a
commit 11f388781c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 57 additions and 0 deletions

View File

@ -0,0 +1,57 @@
# Black Re-ID: A Head-shoulder Descriptor for the Challenging Problem of Person Re-Identification
## Training
To train a model, run
```bash
CUDA_VISIBLE_DEVICES=gpus python train_net.py --config-file <config.yml>
```
## Evaluation
To evaluate the model in test set, run similarly:
```bash
CUDA_VISIBLE_DEVICES=gpus python train_net.py --config-file <configs.yaml> --eval-only MODEL.WEIGHTS model.pth
```
## Experimental Results
### Market1501 dataset
| Method | Pretrained | Rank@1 | mAP |
| :---: | :---: | :---: |:---: |
| ResNet50 | ImageNet | 93.3% | 84.6% |
| MGN | ImageNet | 95.7% | 86.9% |
| HAA (ResNet50) | ImageNet | 95% | 87.1% |
| HAA (MGN) | ImageNet | 95.8% | 89.5% |
### DukeMTMC dataset
| Method | Pretrained | Rank@1 | mAP |
| :---: | :---: | :---: |:---: |
| ResNet50 | ImageNet | 86.2% | 75.3% |
| MGN | ImageNet | 88.7% | 78.4% |
| HAA (ResNet50) | ImageNet | 87.7% | 75.7% |
| HAA (MGN) | ImageNet | 89% | 80.4% |
### Black-reid black group
| Method | Pretrained | Rank@1 | mAP |
| :---: | :---: | :---: |:---: |
| ResNet50 | ImageNet | 80.9% | 70.8% |
| MGN | ImageNet | 86.7% | 79.1% |
| HAA (ResNet50) | ImageNet | 86.7% | 79% |
| HAA (MGN) | ImageNet | 91.0% | 83.8% |
### White_reid whilte group
| Method | Pretrained | Rank@1 | mAP |
| :---: | :---: | :---: |:---: |
| ResNet50 | ImageNet | 89.5% | 75.8% |
| MGN | ImageNet | 94.3% | 85.8% |
| HAA (ResNet50) | ImageNet | 93.5% | 84.4% |
| HSE (MGN) | ImageNet | 95.3% | 88.1% |