Function test_graph

tests/bin/test_mxnet_mobileface.cpp:62–124

}

int test_graph(graph_t graph, const char* dump_file, const float means[], const float scale)
{
    /* get input tensor */
    tensor_t input_tensor = get_graph_input_tensor(graph, 0, 0);
    if(input_tensor == nullptr)
    {
        std::printf("Cannot find input tensor of graph\n");
        return -1;
    }

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

    /* setup input buffer */
    float* input_data = ( float* )malloc(sizeof(float) * img_h * img_w * 3);
    get_input_data(image_file, input_data, img_h, img_w, means, scale);

    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 the graph */
    int ret_prerun = prerun_graph(graph);
    if(ret_prerun < 0)
    {
        std::printf("prerun failed\n");
        return -1;
    }
    // dump_graph(graph);

    unsigned long start_time = get_cur_time();

    for(int i = 0; i < repeat_count; i++)
        run_graph(graph, 1);

    unsigned long end_time = get_cur_time();

    unsigned long off_time = end_time - start_time;
    std::printf("Repeat [%d] time %.2f us per RUN. used %lu us\n", repeat_count, 1.0f * off_time / repeat_count,
                off_time);

    /* get output tensor */
    tensor_t output_tensor = get_graph_output_tensor(graph, 0, 0);
    if(output_tensor == nullptr)
    {
        std::printf("Cannot find output tensor\n");
        return -1;
    }

    int size = get_tensor_buffer_size(output_tensor);
    std::printf("output tensor buffer size: %d\n", size);

    float* data1 = ( float* )get_tensor_buffer(output_tensor);

    dump_float(dump_file, data1, size / 4);