search

tinyvision / SOLIDER-REID

MIT License
55 stars 12 forks source link

Imagenet pre-trained weights #15

Open VlSomers opened 10 months ago

VlSomers commented 10 months ago

Dear authors, thank you for this amazing work.

I'm currently using your codebase, but I would like to use the imagenet SwinTransformer weights as pre-training, instead of the Swin-SOLIDER weights. For this purpose, I'm setting and . Where can I download the imagenet pre-trained weights that will work with your codebase?

From what I understood, I should go to the original SwinFormer repository (https://github.com/microsoft/Swin-Transformer) and download the weights there. Which weights should I download?

Thank you for your help!

cwhgn commented 10 months ago

We follow the swin implementation from mmdet, and download the swin model from official website. The mmdet swin implementation can load the official-website model as long as you set the convert_weights=True. More details can be found in mmdet.