add samples

wangrong

2025-03-19 291afb9c81aae1d15d9e3edcea618d0a2e1fc5a6

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#include "../nodes/vp_app_src_node.h" 
#include "../nodes/infers/vp_ppocr_text_detector_node.h" 
#include "../nodes/osd/vp_text_osd_node.h" 
#include "../nodes/vp_screen_des_node.h" 
#include "../utils/analysis_board/vp_analysis_board.h" 
 
/* 
* ## app src sample ## 
* receive frames(cv::Mat) from host code 
*/ 
 
int main() { 
    VP_SET_LOG_LEVEL(vp_utils::vp_log_level::INFO); 
    VP_LOGGER_INIT(); 
 
    // create nodes 
    auto app_src_0 = std::make_shared<vp_nodes::vp_app_src_node>("app_src_0", 0); 
    auto ppocr_text_detector = std::make_shared<vp_nodes::vp_ppocr_text_detector_node>("ppocr_text_detector",  
                                "./vp_data/models/text/ppocr/ch_PP-OCRv3_det_infer", 
                                "./vp_data/models/text/ppocr/ch_ppocr_mobile_v2.0_cls_infer", 
                                "./vp_data/models/text/ppocr/ch_PP-OCRv3_rec_infer", 
                                "./vp_data/models/text/ppocr/ppocr_keys_v1.txt"); 
    auto osd_0 = std::make_shared<vp_nodes::vp_text_osd_node>("osd_0", "./vp_data/font/NotoSansCJKsc-Medium.otf"); 
    auto screen_des_0 = std::make_shared<vp_nodes::vp_screen_des_node>("screen_des_0", 0); 
 
    // construct pipeline 
    ppocr_text_detector->attach_to({app_src_0}); 
    osd_0->attach_to({ppocr_text_detector}); 
    screen_des_0->attach_to({osd_0}); 
 
    // start pipeline 
    app_src_0->start(); 
 
    // visualize pipeline for debug 
    vp_utils::vp_analysis_board board({app_src_0}); 
    board.display(1, false); // no block since we need interactions from console later 
 
    // simulate push frame to pipeline regularly in a separate thread 
    bool exit = false; 
    auto simulate_run = [&]() { 
        auto index = 0; 
        auto count = 0; 
        auto path = "./vp_data/test_images/text/"; 
        while (!exit) { 
            auto frame = cv::imread(path + std::to_string(index) + ".jpg"); 
            assert(!frame.empty()); 
             
            app_src_0->push_frames({frame});  // push frame to pipeline, return false means failed 
            count++; 
            std::cout << "main thread has pushed [" << count << "] frames into pipeline..." << std::endl; 
             
            index++; 
            index = index % 3; 
            std::this_thread::sleep_for(std::chrono::milliseconds(2000));  // sleep for 2 seconds 
        } 
    }; 
    std::thread simulate_thread(simulate_run); 
 
    /* interact from console, start or stop the pipeline */ 
    /* no except check */ 
    std::string input; 
    std::getline(std::cin, input); 
    while (input != "quit") { 
        if (input == "start") { 
            app_src_0->start(); 
        } 
        else if (input == "stop") { 
            app_src_0->stop();  // app_src_0->push_frames(...) will print Warn message since it has stopped working 
        } 
        else { 
            std::cout << "invalid command!" << std::endl; 
        } 
        std::getline(std::cin, input); 
    } 
 
    std::cout << "app_src_sample sample exits..." << std::endl; 
    exit = true; 
    simulate_thread.join(); 
    app_src_0->detach_recursively(); 
}