| import timm | |
| from .swin_common import _make_swin_backbone | |
| def _make_pretrained_swin2l24_384(pretrained, hooks=None): | |
| model = timm.create_model("swinv2_large_window12to24_192to384_22kft1k", pretrained=pretrained) | |
| hooks = [1, 1, 17, 1] if hooks == None else hooks | |
| return _make_swin_backbone( | |
| model, | |
| hooks=hooks | |
| ) | |
| def _make_pretrained_swin2b24_384(pretrained, hooks=None): | |
| model = timm.create_model("swinv2_base_window12to24_192to384_22kft1k", pretrained=pretrained) | |
| hooks = [1, 1, 17, 1] if hooks == None else hooks | |
| return _make_swin_backbone( | |
| model, | |
| hooks=hooks | |
| ) | |
| def _make_pretrained_swin2t16_256(pretrained, hooks=None): | |
| model = timm.create_model("swinv2_tiny_window16_256", pretrained=pretrained) | |
| hooks = [1, 1, 5, 1] if hooks == None else hooks | |
| return _make_swin_backbone( | |
| model, | |
| hooks=hooks, | |
| patch_grid=[64, 64] | |
| ) | |