Function inference

yolox_example/main.cpp:370–398

}

std::vector<float> inference(const char* model_path, std::vector<float>& input_data) {
    LITE_set_log_level(INFO);
    LiteNetwork model;
    LiteTensor input;
    size_t input_length;
    //! make network
    LITE_CAPI_CHECK(
            LITE_make_network(&model, *default_config(), *default_network_io()),
            "create model error. \n");
    LITE_CAPI_CHECK(
            LITE_load_model_from_path(model, model_path), "load model error. \n");
    //! set input tensor
    LITE_CAPI_CHECK(
            LITE_get_io_tensor(model, "data", LITE_INPUT, &input),
            "get input tensor failed\n");
    LITE_CAPI_CHECK(
            LITE_get_tensor_total_size_in_byte(input, &input_length),
            "get input tensor size failed\n");

    LITE_CAPI_CHECK(
            LITE_reset_tensor_memory(input, input_data.data(), input_length),
            "set input ptr failed\n");

    LITE_CAPI_CHECK(LITE_destroy_tensor(input), "destory input tensor");

    std::vector<float> output = run_model(model);
    LITE_CAPI_CHECK(LITE_destroy_network(model), "delete model failed\n");
    return output;
}

std::string keys =
        "{ @model          | | the inference model path}"