2. 设置预测模型

2.1. 从文件中加载预测模型

API定义如下:

// 设置模型文件路径
// 参数:model - 模型文件所在路径
//      params - 模型参数文件所在路径
// 返回:None
func (config *Config) SetModel(model, params string)

// 获取 Combined 模型的模型文件路径
// 参数:无
// 返回:string - 模型文件路径
func (config *Config) ProgFile() string

// 获取 Combined 模型的参数文件路径
// 参数:无
// 返回:string - 参数文件路径
func (config *Config) ParamsFile() string

代码示例:

package main

// 引入 Paddle Golang Package
import pd "github.com/paddlepaddle/paddle/paddle/fluid/inference/goapi"
import fmt

func main() {
    // 创建 Config 对象
    config := paddle.NewConfig()

    // 设置预测模型路径,这里为 Combined 模型
    config.SetModel("data/resnet.pdmodel", "data/resnet.pdiparams")

    // 输出模型路径
    fmt.Println("Combined model path is: ", config.ProgFile())
    fmt.Println("Combined param path is: ", config.ParamsFile())
}