Function main

tests/bin/vgg16.cpp:56–166

}

int main(int argc, char* argv[])
{
    int img_h = 224;
    int img_w = 224;

    /* prepare input data */
    float* input_data = ( float* )malloc(sizeof(float) * img_h * img_w * 3);

    init_tengine();
    if(request_tengine_version("0.9") < 0)
        return 1;

    graph_t graph = create_graph(nullptr, "caffe", text_file, model_file);
    if(graph == nullptr)
    {
        std::cout << "Create graph failed\n";
        std::cout << "errno: " << get_tengine_errno() << "\n";
        return 1;
    }

    /* run the graph */
    prerun_graph(graph);

    int node_idx = 0;
    int tensor_idx = 0;

    const char* repeat_count = std::getenv("REPEAT_COUNT");
    if(repeat_count)
        repeat = std::strtoul(repeat_count, NULL, 10);

    for(int i = 0; i < repeat; i++)
    {
        /* get input tensor */
        get_input_data(image_file, input_data, img_h, img_w, channel_mean, 1);

        tensor_t input_tensor = get_graph_input_tensor(graph, node_idx, tensor_idx);

        if(input_tensor == nullptr)
        {
            std::printf("Cannot find input tensor,node_idx: %d,tensor_idx: %d\n", node_idx, tensor_idx);
            return -1;
        }

        int dims[] = {1, 3, img_h, img_w};
        set_tensor_shape(input_tensor, dims, 4);

        /* setup input buffer */
        if(set_tensor_buffer(input_tensor, input_data, 3 * img_h * img_w * 4) < 0)
        {
            std::printf("Set buffer for tensor failed\n");
            return -1;
        }

        run_graph(graph, 1);

        release_graph_tensor(input_tensor);
    }