pytorch转成paddle模型
等待自然人 发布于2021-11 浏览:3366 回复:1
0
收藏

将pytorch先转成onnx模型然后再转换成paddle模型得到model.pdiparams,model.pdiparams.info,model.pdmodel三个文件,但是在Edgeboard上进行部署的时候预测报错,这是什么原因,在转成三个文件后还需要进行什么操作吗

收藏
点赞
0
个赞
共1条回复 最后由wano43回复于2022-03
#2回憶那纯真温馨回复于2022-02

我模仿demo写了一段加载这种模型的代码,但是运行的时候会报Segment Fault,分享给你看能否加载pdiparams这种参数。

import numpy as np
from edgeboard import *
from core import *
from PIL import Image

#1 加载模型
predictor = PaddleLitePredictor()
predictor.set_model_file('model/mymodel.pdmodel')
predictor.set_param_file('model/mymodel.pdiparams')
try:
    predictor.load()
    print("Predictor Init Success !!!")
except:
    print("Error: CreatePaddlePredictor Failed.")

#2 读取待预测图片
image = Image.open('./num.jpg')
#对图片进行预处理
image = np.array(image)
image = image/127.5 - 1
#3 设置输入数据
# 获取第0位置处的输入空tensor(类似指针),预处理后的结果最终会拷贝到这个tensor中
input_tensor = predictor.get_input(0)
input_tensor.from_numpy(image)

#4 执行预测
predictor.run()

#5 得到输出结果
predict_result = predictor.get_output(0)
print(predict_result.shape())
print(predict_result.numpy())
0
TOP
切换版块