Exemplo n.º 1
0
 def __init__(self):
     backbone = resnet152(pretrained=True, strides=(2, 2, 2, 1))
     super(ResNet152EdgeDisplacement, self).__init__(backbone=backbone)
Exemplo n.º 2
0
 def __init__(self, num_classes=20):
     backbone = resnet152(pretrained=True, strides=(2, 2, 2, 1))
     conv_output = 1024
     super(ResNet152, self).__init__(backbone=backbone,
                                     num_classes=num_classes,
                                     conv_output=conv_output)
Exemplo n.º 3
0
 def __init__(self, path_index):
     backbone = resnet152(pretrained=True, strides=(2, 2, 2, 1))
     super(ResNet152AffinityDisplacementLoss,
           self).__init__(path_index, backbone=backbone)