def enet_tusimple(num_classes, encoder_only, continue_from):
return enet_(
num_classes=num_classes,
num_lanes=num_classes - 1,
dropout_1=0.01,
dropout_2=0.1,
flattened_size=4400,
encoder_only=encoder_only,
pretrained_weights=continue_from if not encoder_only else None)
def enet(num_classes, encoder_only, continue_from):
# Define ENet (Without ImageNet pretraining)
return enet_(num_classes=num_classes, encoder_only=encoder_only,
pretrained_weights=continue_from if not encoder_only else None)