search

ZhangGongjie / SAM-DETR

[CVPR'2022] SAM-DETR & SAM-DETR++: Official PyTorch Implementation
MIT License
292 stars 50 forks source link

swin transformer backbone #11

Open zhenshen-mla opened 1 year ago

zhenshen-mla commented 1 year ago

hi. Can I use Swin Transformer as a backbone instead of resnet50? If so, what changes should be made to the swin transformer(pretrained on imgnet22k) ?

ZhangGongjie commented 1 year ago

Sure you can replace resnet50 with Swin! The only thing to change is the backbone-related codes. You may consider to incorporate codes from https://github.com/microsoft/Swin-Transformer .