def __init__(self):
backbone = resnet50(pretrained=True, strides=(2, 2, 2, 1))
super(ResNet50EdgeDisplacement, self).__init__(backbone=backbone)
def __init__(self, num_classes=20):
backbone = resnet50(pretrained=True, strides=(2, 2, 2, 1))
conv_output = 1024
super(ResNet50, self).__init__(backbone=backbone,
num_classes=num_classes,
conv_output=conv_output)
def __init__(self, path_index):
backbone = resnet50(pretrained=True, strides=(2, 2, 2, 1))
super(ResNet50AffinityDisplacementLoss,
self).__init__(path_index, backbone=backbone)