Python erfnet_resnet Examples

Example #1

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)

Example #2

File: all_utils_landec.py Project: ozcell/pytorch-auto-drive

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)

Example #3

File: all_utils_semseg.py Project: noticeable/pytorch-auto-drive

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)