Spaces:
Running
Running
from isegm.utils.serialization import serialize | |
from .is_model import ISModel | |
from .modeling.swin_transformer import SwinTransformer, SwinTransfomerSegHead | |
class SwinformerModel(ISModel): | |
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} |