Spaces:
Running
Running
File size: 677 Bytes
6d1366a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from isegm.utils.serialization import serialize
from .is_model import ISModel
from .modeling.swin_transformer import SwinTransformer, SwinTransfomerSegHead
class SwinformerModel(ISModel):
@serialize
def __init__(
self,
backbone_params={},
head_params={},
**kwargs
):
super().__init__(**kwargs)
self.backbone = SwinTransformer(**backbone_params)
self.head = SwinTransfomerSegHead(**head_params)
def backbone_forward(self, image, coord_features=None):
backbone_features = self.backbone(image, coord_features)
return {'instances': self.head(backbone_features), 'instances_aux': None} |