search

VlSomers / bpbreid

[WACV23] A strong baseline for body part-based person re-identification
Other
166 stars 16 forks source link

shape error #30

Closed 1024AILab closed 1 year ago

1024AILab commented 1 year ago

I follow the instruction as you say, but I can't run the code image

TInaWangxue commented 1 year ago

I also met this issue. The pretrained model I used is "bpbreid_occluded_duke_hrnet32_10670.pth". Actually I dont know the size of the masks which this project provided. I don't generate masks by myself.

VlSomers commented 1 year ago

Hi, this is an error with the new version of Albumentation, I just added a new commit to fix this, you can pull the last version. In "torchreid/data/transforms.py", I had to add the following argument to Compose(...): "is_check_shapes=False"

VlSomers commented 1 year ago

Please let me know if it works

1024AILab commented 1 year ago

281253383-0234850d-de4e-4b6a-b475-452294313c0d I changed 'is_check_shape' to 'False' in composition.py, and it ran successfully. However, I haven't tried the new code yet.

TInaWangxue commented 1 year ago

Please let me know if it works

It works! Thank you so much!

1024AILab commented 1 year ago

@TInaWangxue You are welcome! Are you Chinese? I'm ungraduate and very interested in person reid.

TInaWangxue commented 1 year ago

@TInaWangxue You are welcome! Are you Chinese? I'm ungraduate and very interested in person reid.

Yes, I'm a Chinese graduate student. Recently I've been looking at work relating to person re-identification. I'm so interested in your work.

1024AILab commented 1 year ago

@TInaWangxue I had done some research in short-term reid and Cloth Changing ReID. You can add my wechat 13982058834 and then we can have a deep communication.