1. Config 类定义

Config 类为用于配置构建 Predictor 对象的配置信息,如模型路径、是否开启 gpu 等等。

构造函数定义如下:

# Config 类定义,输入为 None
class paddle.inference.Config()

# Config 类定义,输入为其他 Config 对象
class paddle.inference.Config(config: Config)

# Config 类定义,输入分别为模型文件路径和参数文件路径
class paddle.inference.Config(prog_file: str, params_file: str)

代码示例:

# 引用 paddle inference 预测库
import paddle.inference as paddle_infer

# 创建 config
config = paddle_infer.Config("./mobilenet.pdmodel", "./mobilenet.pdiparams")

# 根据 config 创建 predictor
predictor = paddle_infer.create_predictor(config)