huanghoujing / pal-mgn Goto Github PK

Hello,
I'm trying to run the project but I'm unable to find some required files, although I think I downloaded all the files from Google drive.
In particular, if I run:
gpus=0 python_exc=python train_set=msmt17 testset_names=msmt17 only_test=True bash train_mgn.sh
I obtain:
No such file or directory: '.../msmt17/MSMT17_V1/list_train.txt'
I was not able to find list_train.txt within the downloaded files.

If I run instead:
gpus=0 python_exc=python train_set=market1501 testset_names=market1501 only_test=True bash train_mgn.sh
I obtain:
model_weight_file exp/train_mgn/market1501/model_weight.pth does not exist!
The same applies for cuhk03 and duke.

I see lot of ckpt.pth files in the \exp\eanet\test_paper_models\ subdirectories, but even if I modify the name into model_weight.pth and copy it in the right directory, it does not work as well (unexpected key(s) in state dict). I also tried to use model_weight.pth for MSMT17->MSMT17 on Google drive but I get size mismatches when I load the weight.

Am I doing something wrong?
Thanks

我按照您的readme在market1501数据集上训练和测试了PS-MGN，map仅仅只有0.31，让我百思不得其解。两个数据集的位置也按照您的格式放好了。希望您百忙之中可以帮我解答，谢谢您