aikenml's picture
Upload folder using huggingface_hub
c985ba4
raw
history blame
203 Bytes
from networks.decoders.fpn import FPNSegmentationHead
def build_decoder(name, **kwargs):
if name == 'fpn':
return FPNSegmentationHead(**kwargs)
else:
raise NotImplementedError