Skip to content

Latest commit

 

History

History
67 lines (59 loc) · 2.48 KB

README.md

File metadata and controls

67 lines (59 loc) · 2.48 KB
Raw

ReID-PCB_RPP

This repository tries to implement the paper:Beyond Part Models: Person Retrieval with Refined Part Pooling and I almost follow the training details at the origin paper.

Market-1501 Results

  • The result of PCB is little higher than paper.
  • The result of PCB+RPP is little lower than paper.
  • I think if we do more augmentation with the training data, the performance will be better.
Feature mAP(%) Rank-1(%)
PCB(paper) G 77.4 92.3
PCB(paper) H 77.3 92.4
PCB(ours) G 78.6 92.7
PCB(ours) H 78.5 92.1
PCB+RPP(paper) G 81.6 93.8
PCB+RPP(paper) H 81.0 93.1
PCB+RPP(ours) G 80.7 92.8
PCB+RPP(ours) H 79.8 92.4

Prerequisites

  • Python 2.7 or 3.5
  • Pytorch 0.4
  • GPU Memory

Getting started

Dataset

  • Dwonload Market1501 Dataset and extract the files to the current folder
  • Change the download_path in the prepare.py and run python prepare.py to prepare the dataset

Train

Train the model:

python train.py --gpu_ids 0,1 --batchsize 64 --data_dir your_data_path --save_dir your_model_save_path --RPP True
  • --gpu_ids: single GPU or mutil-GPUs
  • --batchsize: batch size
  • --data_dir: the path of the preparing data
  • --save_dir: the path to save the training model
  • --RPP: whether to train with RPP I train the model on the two GPUs with 64 batchsize

Test

Test the model(extract the features):

python test.py --gpu_ids 0 --which_epoch select the model --stage PCB or full --RPP True or False --feature_H True or False
  • --gpu_ids: just single GPU
  • --which_epoch: select the i-th model
  • --stage: select the training model : PCB or full(PCB+RPP)
  • --RPP: if you choose the PCB model, RPP to False; if you choose the full model, RPP to True
  • --feature_H: whether to extract the low-dims features

Evaluate

Evaluate the model

  • Evaluate on GPU:
python evaluate_gpu.py --gpu_ids 0 --reslut_mat the path of  the features mat
  • Evaluate on CPU:
python evaluate_gpu.py --reslut_mat the path of  the features mat

Reference resources

Thanks to the layumi/Person_reID_baseline_pytorch. This repository only implements the part of PCB, I make some modifications on it and then add the RPP part.