Method run_ONet

examples/mtcnn/mtcnn.cpp:234–324

}

int mtcnn::run_ONet(image img, std::vector<face_box>& rnet_boxes, std::vector<face_box>& output_boxes)
{
    int batch = rnet_boxes.size();

    int channel = 3;
    int height = 48;
    int width = 48;
    tensor_t input_tensor = get_graph_tensor(ONet_graph, "data");
    int dims[] = {batch, channel, height, width};
    set_tensor_shape(input_tensor, dims, 4);
    int img_size = channel * height * width;
    int in_mem = sizeof(float) * batch * img_size;
    float* input_data = ( float* )malloc(in_mem);
    float* input_ptr = input_data;
    static bool first_run = true;

    set_tensor_buffer(input_tensor, input_ptr, in_mem);
    for(int i = 0; i < batch; i++)
    {
        copy_one_patch(img, rnet_boxes[i], input_ptr, width, height);

        input_ptr += img_size;
    }

    if(first_run)
    {
        if(prerun_graph(ONet_graph) != 0)
        {
            std::cout << "Prerun ONet graph failed, errno: " << get_tengine_errno() << "\n";
            return 1;
        }
        first_run = false;
    }

    if(run_graph(ONet_graph, 1) != 0)
    {
        std::cout << "Run ONet graph failed, errno: " << get_tengine_errno() << "\n";
        return 1;
    }
    free(input_data);
    /* output */
    tensor_t tensor = get_graph_tensor(ONet_graph, "conv6-3");
    float* points_data = ( float* )get_tensor_buffer(tensor);

    tensor = get_graph_tensor(ONet_graph, "prob1");
    float* confidence_data = ( float* )get_tensor_buffer(tensor);

    tensor = get_graph_tensor(ONet_graph, "conv6-2");
    float* reg_data = ( float* )get_tensor_buffer(tensor);

    int conf_page_size = 2;
    int reg_page_size = 4;
    int points_page_size = 10;
    for(int i = 0; i < batch; i++)
    {
        if(*(confidence_data + 1) > conf_o_threshold_)
        {
            face_box output_box;