Exemple #1
0
def rx101_unet(num_classes, num_channels=3, pretrained=True):
    return n05_selim_zoo.resnet_unet(seg_classes=num_classes,
                                     backbone_arch='resnext101')
Exemple #2
0
def pd_rn154_unet(num_classes, num_channels=3, pretrained=True):
    return n05_selim_zoo.resnet_unet(seg_classes=num_classes,
                                     backbone_arch='pd_resnet154')