File size: 203 Bytes
c985ba4
 
 
 
 
 
 
 
 
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