def create_model(model_class, backbone):
    """Initializes a segmentation model.

    Args:
        model_class: One of smp.Unet, smp.UnetPlusPlus
        backbone: Encoder backbone. Should always be "mobilenet_v2"
                    for this purpose.

    """
    model = model_class(
        encoder_name=backbone, encoder_weights=None, in_channels=3, classes=2
    )
    return model