Commit 62415db5 authored by Your Name's avatar Your Name
Browse files

修改模型输入数据格式

parent faeb4c4a
...@@ -25,7 +25,7 @@ class Crnn: ...@@ -25,7 +25,7 @@ class Crnn:
inputImage = self.prepare_input(image) inputImage = self.prepare_input(image)
# 执行推理 # 执行推理
results = self.model.run({self.model.get_parameter_names()[0]: migraphx.argument(inputImage)}) results = self.model.run({self.model.get_parameter_names()[0]: inputImage})
# 获取第一个输出节点的数据,migraphx.argument类型 # 获取第一个输出节点的数据,migraphx.argument类型
result=results[0] result=results[0]
result=np.array(result) result=np.array(result)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment