def __init__(self,
nclass,
aux=False,
backbone='shufflenet',
pretrained_base=False,
**kwargs):
super(ShuffleNetSeg, self).__init__(nclass, aux, backbone,
pretrained_base, **kwargs)
self.head = _FCNHead(1536, nclass, **kwargs)
if aux:
self.auxlayer = _FCNHead(768, nclass, **kwargs)
def __init__(self,
nclass,
aux=False,
backbone='efficientnet',
pretrained_base=False,
**kwargs):
super(EfficientNetSeg, self).__init__(nclass, aux, backbone,
pretrained_base, **kwargs)
self.head = _FCNHead(320, nclass, **kwargs)
if aux:
self.auxlayer = _FCNHead(112, nclass, **kwargs)
def __init__(self,
nclass,
aux=False,
backbone='mobilenet',
pretrained_base=False,
**kwargs):
super(MobileNetSeg, self).__init__(nclass, aux, backbone,
pretrained_base, **kwargs)
self.head = _FCNHead(1024, nclass, **kwargs)
if self.aux:
self.auxlayer = _FCNHead(512, nclass, **kwargs)
def __init__(self, nclass, aux=False, backbone='mobilenetv2', pretrained_base=False, **kwargs):
super(IGCV3Seg, self).__init__(nclass, aux, backbone, pretrained_base, **kwargs)
self.head = _Head(nclass, **kwargs)
if self.aux:
self.auxlayer = _FCNHead(160, nclass, **kwargs)