Python get_modelの例

コード例 #1

def prepare_model(model_name, use_pretrained, pretrained_model_file_path):
    kwargs = {"pretrained": use_pretrained}

    net = get_model(model_name, **kwargs)

    if pretrained_model_file_path:
        assert (os.path.isfile(pretrained_model_file_path))
        logging.info("Loading model: {}".format(pretrained_model_file_path))
        net.load_weights(filepath=pretrained_model_file_path)

    return net

コード例 #2

ファイル: utils.py プロジェクト: jdc08161063/imgclsmob

def prepare_model(model_name, classes, use_pretrained,
                  pretrained_model_file_path):
    kwargs = {'pretrained': use_pretrained, 'classes': classes}

    net = get_model(model_name, **kwargs)

    if pretrained_model_file_path:
        assert (os.path.isfile(pretrained_model_file_path))
        logging.info('Loading model: {}'.format(pretrained_model_file_path))
        net.load_weights(filepath=pretrained_model_file_path)

    return net