修改input/output代码

Doc/Tutorial_Cpp.md

@@ -23,10 +23,21 @@ ErrorCode Unet::Initialize(InitializationParameterOfSegmentation initParamOfSegm
     net = migraphx::parse_onnx(modelPath); 
     LOG_INFO(logFile,"succeed to load model: %s\n",GetFileName(modelPath).c_str());
 
-    // 获取模型输入属性
-    std::pair<std::string, migraphx::shape> inputAttribute=net.get_parameter_shapes();
-    inputName=inputAttribute.first;
-    inputShape=inputAttribute.second;
+    // 获取模型输入/输出节点信息
+    std::cout<<"inputs:"<<std::endl;
+    std::unordered_map<std::string, migraphx::shape> inputs=net.get_inputs();
+    for(auto i:inputs)
+    {
+        std::cout<<i.first<<":"<<i.second<<std::endl;
+    }
+    std::cout<<"outputs:"<<std::endl;
+    std::unordered_map<std::string, migraphx::shape> outputs=net.get_outputs();
+    for(auto i:outputs)
+    {
+        std::cout<<i.first<<":"<<i.second<<std::endl;
+    }
+    inputName=inputs.first;
+    inputShape=inputs.second;
     int N=inputShape.lens()[0];
     int C=inputShape.lens()[1];
     int H=inputShape.lens()[2];

Python/Unet.py

@@ -26,8 +26,20 @@ if __name__ == '__main__':
     
     # 加载模型
     model = migraphx.parse_onnx("../Resource/Models/unet_13_256.onnx")
+
+    # 获取模型输入/输出节点信息
+    print("inputs:")
+    inputs = model.get_inputs()
+    for key,value in inputs.items():
+        print("{}:{}".format(key,value))
+
+    print("outputs:")
+    outputs = model.get_outputs()
+    for key,value in outputs.items():
+        print("{}:{}".format(key,value))
+
     inputName = model.get_parameter_names()
-    inputShape = model.get_parameter_shapes()
+    inputShape = inputs[inputName].lens()
     print("inputName:{0} \ninputShape:{1}".format(inputName, inputShape))
 
     # 编译模型

Src/Unet.cpp

@@ -53,10 +53,21 @@ ErrorCode Unet::Initialize(InitializationParameterOfSegmentation initParamOfSegm
     net = migraphx::parse_onnx(modelPath);       
     LOG_INFO(stdout,"succeed to load model: %s\n",GetFileName(modelPath).c_str());
 
-    // 获取模型输入属性
-    std::unordered_map<std::string, migraphx::shape> inputMap=net.get_parameter_shapes();
-    inputName=inputMap.begin()->first;
-    inputShape=inputMap.begin()->second;
+    // 获取模型输入/输出节点信息
+    std::cout<<"inputs:"<<std::endl;
+    std::unordered_map<std::string, migraphx::shape> inputs=net.get_inputs();
+    for(auto i:inputs)
+    {
+        std::cout<<i.first<<":"<<i.second<<std::endl;
+    }
+    std::cout<<"outputs:"<<std::endl;
+    std::unordered_map<std::string, migraphx::shape> outputs=net.get_outputs();
+    for(auto i:outputs)
+    {
+        std::cout<<i.first<<":"<<i.second<<std::endl;
+    }
+    inputName=inputs.begin()->first;
+    inputShape=inputs.begin()->second;
     int N=inputShape.lens()[0];
     int C=inputShape.lens()[1];
     int H=inputShape.lens()[2];