first commit
This commit is contained in:
13
midas/backbones/swin.py
Normal file
13
midas/backbones/swin.py
Normal file
@@ -0,0 +1,13 @@
|
||||
import timm
|
||||
|
||||
from .swin_common import _make_swin_backbone
|
||||
|
||||
|
||||
def _make_pretrained_swinl12_384(pretrained, hooks=None):
|
||||
model = timm.create_model("swin_large_patch4_window12_384", pretrained=pretrained)
|
||||
|
||||
hooks = [1, 1, 17, 1] if hooks == None else hooks
|
||||
return _make_swin_backbone(
|
||||
model,
|
||||
hooks=hooks
|
||||
)
|
||||
Reference in New Issue
Block a user