deep-person-reid icon indicating copy to clipboard operation
deep-person-reid copied to clipboard

How to apply cross dataset validation on custom torchreid model

Open FatimaZulfiqar opened this issue 3 years ago • 4 comments

Hi. I have a custom person reid model that is currently not available in this torchreid library. I want to perform cross dataset validation on that custom-designed model. The model is originally trained on msmt17 dataset and I have its checkpoint saved. I want to evaluate this model on market1501 or dukemtmc. How can I do that? Can anybody please guide me?

FatimaZulfiqar avatar Jun 27 '21 21:06 FatimaZulfiqar

https://github.com/KaiyangZhou/deep-person-reid#cross-domain-setting

KaiyangZhou avatar Jun 28 '21 03:06 KaiyangZhou

Thank you for your response. yes, I did that but the issue is that the person Reid model that I have is not present in this GitHub repository. I have its weights saved in pytorch format. I want to load the model weights and then perform cross dataset validation

FatimaZulfiqar avatar Jun 28 '21 08:06 FatimaZulfiqar

To use your own model, you can simply put your model in the initialization of the engine (see https://github.com/KaiyangZhou/deep-person-reid#get-started-30-seconds-to-torchreid)

KaiyangZhou avatar Jun 29 '21 04:06 KaiyangZhou

If you are referring to model weights, simply load the weights to the specific model before giving it to the engine

see https://github.com/KaiyangZhou/deep-person-reid/blob/master/scripts/main.py#L168

KaiyangZhou avatar Jun 29 '21 04:06 KaiyangZhou