修改input/output代码

Python/RetinaFace_infer_migraphx.py

@@ -59,8 +59,20 @@ if __name__ == '__main__':
         cfg = cfg_re50
     device = torch.device("cpu" if args.cpu else "cuda")
     model = migraphx.parse_onnx("./FaceDetector.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()[0]
-    inputShape=model.get_parameter_shapes()[inputName].lens()
+    inputShape=inputs[inputName].lens()
     print("inputName:{0} \ninputShape:{1}".format(inputName,inputShape))
 
     # FP16

Src/RetinaFace.cpp

@@ -71,10 +71,21 @@ ErrorCode DetectorRetinaFace::Initialize(InitializationParameterOfDetector initi
     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];