1. Config 类定义¶
Config
类为用于配置构建 Predictor
对象的配置信息,如模型路径、是否开启gpu等等。
构造函数定义如下:
# Config 类定义,输入为 None
class paddle.inference.Config()
# Config 类定义,输入为其他 Config 对象
class paddle.inference.Config(config: Config)
# Config 类定义,输入为非 Combine 模型的文件夹路径
class paddle.inference.Config(model_dir: str)
# Config 类定义,输入分别为 Combine 模型的模型文件路径和参数文件路径
class paddle.inference.Config(prog_file: str, params_file: str)
代码示例 (1):加载预测模型 - 非Combined模型
# 引用 paddle inference 预测库
import paddle.inference as paddle_infer
# 创建 config
config = paddle_infer.Config()
# 加载非Combined模型
config.set_model("./mobilenet_v1")
# 根据 config 创建 predictor
predictor = paddle_infer.create_predictor(config)
代码示例 (2):加载预测模型 - 非Combined模型
# 引用 paddle inference 预测库
import paddle.inference as paddle_infer
# 创建 config
config = paddle_infer.Config("./mobilenet_v1")
# 根据 config 创建 predictor
predictor = paddle_infer.create_predictor(config)
代码示例 (3):加载预测模型 - Combined模型
# 引用 paddle inference 预测库
import paddle.inference as paddle_infer
# 创建 config
config = paddle_infer.Config("./mobilenet_v2/__model__", "./mobilenet_v2/__params__")
# 根据 config 创建 predictor
predictor = paddle_infer.create_predictor(config)