Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

The training settings and teacher models of training in DukeMTMC and MSMT17. #44

Open
KiritoHugh opened this issue May 19, 2020 · 1 comment

Comments

@KiritoHugh
Copy link

I used the same hyperparameters setting for DukeMTMC and MSMT17 as the setting in Market1501 training config file. And I used the pretrained Resnet model from https://kaiyangzhou.github.io/deep-person-reid/MODEL_ZOO as the teacher models.

Although I got a pretty nice result in Market1501, the result in the other two datasets are really poor(even both poorer than their teacher models).

Market1501:
torch.Size([3368, 1024])
Alpha:0.00 Rank@1:0.9374 Rank@5:0.9860 Rank@10:0.8490 mAP:0.8490
Alpha:0.50 Rank@1:0.9471 Rank@5:0.9872 Rank@10:0.8637 mAP:0.8637
Alpha:-1.00 Rank@1:0.9356 Rank@5:0.9837 Rank@10:0.8048 mAP:0.8048

Duke:
torch.Size([2228, 1024])
Alpha:0.00 Rank@1:0.6683 Rank@5:0.7742 Rank@10:0.8043 mAP:0.4390
Alpha:0.50 Rank@1:0.6849 Rank@5:0.7873 Rank@10:0.8187 mAP:0.4652
Alpha:-1.00 Rank@1:0.6688 Rank@5:0.7733 Rank@10:0.8021 mAP:0.4291

MSMT:
torch.Size([11659, 1024])
Alpha:0.00 Rank@1:0.4731 Rank@5:0.6029 Rank@10:0.6484 mAP:0.2736
Alpha:0.50 Rank@1:0.4754 Rank@5:0.6052 Rank@10:0.6544 mAP:0.2799
Alpha:-1.00 Rank@1:0.3923 Rank@5:0.5260 Rank@10:0.5779 mAP:0.1903

I also post the training loss log of DukeMTMC below.

  • Does the teacher models' ability have a huge effect on the system's final ability?
  • (小声哔哔:Could you share the hyperparameter training settings and teacher model on the other datasets?(T▽T)太穷了,跑一次要花50RMB,调不起参)
  • I also wonder why you 大神们 know how to tune the hyperparameters, where I can learn some tricks about tuning? Only by practicing?

Your reply would be highly appreciated!

DUKE

@layumi
Copy link
Contributor

layumi commented May 19, 2020

Hi @KiritoHugh
Thanks for your attention on our paper.
This is my teacher config based on https://github.com/layumi/Person_reID_baseline_pytorch
DukeMTMC-reID

PCB: false
angle: false
balance: false
batchsize: 8
color_jitter: false
data_dir: ../DukeMTMC-reID/pytorch
droprate: 0.75
erasing_p: 0
fp16: false
gpu_ids: '0'
h: 256
lr: 0.01
name: Duke_batch8_lr0.010000_p0.00_d0.75
stride: 1
train_all: true
use_dense: false
w: 128
pool: avg
nclass: 702

MSMT-17

PCB: false
angle: false
balance: false
batchsize: 8
color_jitter: false
data_dir: ../MSMT17_V1/pytorch
droprate: 0.5
erasing_p: 0
fp16: false
gpu_ids: '0'
h: 256
lr: 0.01
name: MSMT_batch8_lr0.010000_p0.00_d0.50
stride: 1
train_all: true
use_dense: false
w: 128
pool: avg
nclass: 1041

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants