Ort::Session session(env, model_path, session_options);
//打印模型的输入层(node names, types, shape etc.)
Ort::AllocatorWithDefaultOptions allocator;
//输出模型输入节点的数量
size_t num_input_nodes = session.GetInputCount();
size_t num_output_nodes = session
Onnxruntime输入和输出打印
猜你喜欢
转载自blog.csdn.net/weixin_42990464/article/details/126971520
今日推荐
周排行