Imagenet pre-trained weights

[VlSomers]

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 <PRETRAIN_CHOICE = 'imagenet'> and <PRETRAIN_PATH='/path/to/imagenet_pretrained_weights'>. 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?

• Swin-V1 or Swin-V2?
• Swin-B?
• 384x384 or 224x224?

Thank you for your help!

[cwhgn]

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.