layumi / Person_reID_baseline_pytorch

:bouncing_ball_person: Pytorch ReID: A tiny, friendly, strong pytorch implement of person re-id / vehicle re-id baseline. Tutorial 👉https://github.com/layumi/Person_reID_baseline_pytorch/tree/master/tutorial
https://www.zdzheng.xyz
MIT License
4.13k stars 1.01k forks source link

训练后精度不高 #413

Open qianhe-666 opened 2 months ago

qianhe-666 commented 2 months ago

这是我的命令行:python train.py --gpu_ids 1 --name ft_ResNet50 --train_all --batchsize 32 --data_dir /home/user/qh/Person_reID_baseline_pytorch-master/Market/pytorch 训练200轮,之后进行测试python test.py --gpu_ids 0 --name ft_ResNet --test_dir /home/user/qh/Person_reID_baseline_pytorch-master/Market/pytorch --batchsize 32 --which_epoch 200,得出的结果Rank01:0.086995 Ranka5:0.100950 Rank010:0.104513 mAP:0.071516,这个结果精度不太行,我这是问题出现在哪里

layumi commented 2 months ago

@qianhe-666 你好, 你两次 name 不一样。 训练的是 ft_ResNet50, 测试的是 ft_ResNet