获取onnx模型的类别

可以使用ONNX Runtime的API函数获取ONNX模型的类别。以下是使用C++代码示例:

#include <iostream>
#include <string>
#include <vector>
#include "onnxruntime_cxx_api.h"

int main() {
    Ort::Env env(ORT_LOGGING_LEVEL_WARNING, "test");
    Ort::Session session(env, "model.onnx", Ort::SessionOptions());
    Ort::AllocatorWithDefaultOptions allocator;

    // 获取模型的输入名称和形状
    std::vector<const char*> input_node_names = session.GetInputNames();
    std::vector<int64_t> input_node_dims = session.GetInputTypeInfo(0).GetTensorTypeAndShapeInfo().GetShape();

    // 创建输入张量
    Ort::Value input_tensor = Ort::Value::CreateTensor<float>(allocator, input_node_dims.data(), input_node_dims.size());

    // 获取模型的输出名称和形状
    std::vector<const char*> output_node_names = session.GetOutputNames();
    std::vector<int64_t> output_node_dims = session.GetOutputTypeInfo(0).GetTensorTypeAndShapeInfo().GetShape();

    // 创建输出张量
    Ort::Value output_tensor = Ort::Value::CreateTensor<float>(allocator, output_node_dims.data(), output_node_dims.size());

    // 获取模型的类别
    Ort::TypeInfo type_info = session.GetOutputTypeInfo(0);
    const char* type = type_info.GetTensorTypeAndShapeInfo().GetElementType();

    std::cout << "ONNX模型的类别为：" << type << std::endl;

    return 0;
}

其中，"model.onnx"是ONNX模型的文件路径。通过调用session.GetOutputTypeInfo(0)获取输出张量的类型信息，再通过调用GetTensorTypeAndShapeInfo().GetElementType()获取模型的类别。