search

tinyvision / SOLIDER-REID

MIT License
67 stars 12 forks source link

Imagenet pre-trained weights #15

Open VlSomers opened 1 year ago

VlSomers commented 1 year 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 1 year 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.