def erfnet_tusimple(num_classes,
scnn=False,
pretrained_weights='erfnet_encoder_pretrained.pth.tar'):
# Define ERFNet for TuSimple (With only ImageNet pretraining)
return erfnet_resnet(pretrained_weights=pretrained_weights,
num_classes=num_classes,
aux=6,
dropout_1=0.1,
dropout_2=0.1,
flattened_size=2560,
scnn=scnn)
def erfnet_culane(num_classes,
scnn=False,
pretrained_weights='erfnet_encoder_pretrained.pth.tar'):
# Define ERFNet for CULane (With only ImageNet pretraining)
return erfnet_resnet(pretrained_weights=pretrained_weights,
num_classes=num_classes,
num_lanes=num_classes - 1,
dropout_1=0.1,
dropout_2=0.1,
flattened_size=4500,
scnn=scnn)
def erfnet(num_classes,
pretrained_weights='erfnet_encoder_pretrained.pth.tar'):
# Define ERFNet (With only ImageNet pretraining)
return erfnet_resnet(pretrained_weights=pretrained_weights,
num_classes=num_classes)