INNER CODE UNIT · Python

_get_model_instance

meetps/pytorch-semseg · ptsemseg/models/__init__.py:50

def _get_model_instance(name):
    try:
        return {
            "fcn32s": fcn32s,
            "fcn8s": fcn8s,
            "fcn16s": fcn16s,
            "unet": unet,
            "segnet": segnet,
            "pspnet": pspnet,
            "icnet": icnet,
            "icnetBN": icnet,
            "linknet": linknet,
            "frrnA": frrn,
            "frrnB": frrn,
        }[name]
    except:
        raise ("Model {} not available".format(name))

View source record →