Spaces:
Running
Running
File size: 203 Bytes
ffbe0b4 |
1 2 3 4 5 6 7 8 9 10 |
from networks.decoders.fpn import FPNSegmentationHead
def build_decoder(name, **kwargs):
if name == 'fpn':
return FPNSegmentationHead(**kwargs)
else:
raise NotImplementedError
|