search

FZfangzheng / Swin-Transformer-Semantic-Segmentation-Without-mmsegmentation

Unofficial implementation of Swin-Transformer-Semantic-Segmentation, relatively independent code, easy to add to other models.
Apache License 2.0
16 stars 1 forks source link

The question about the type of SwinTransfomer #1

Closed super233 closed 2 years ago

super233 commented 2 years ago

Hi, thanks for your awesome code.

Which type is this SwinTransformer, Swin-T, Swin-S or Swin-B? And how can I modify code to use one of them?

FZfangzheng commented 2 years ago

Hi, this is Swin-B. You can choose different types by modifying the relevant parameters in /swin_transformer/backbones/swin_transformer.py#L478. For parameters, please refer to the configuration in /configs/swin in the official code.

super233 commented 2 years ago

Thanks very much. :-)