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
81
82
83
84
85
#include "../nodes/vp_file_src_node.h" 
#include "../nodes/infers/vp_yolo_detector_node.h" 
#include "../nodes/osd/vp_osd_node.h" 
#include "../nodes/vp_split_node.h" 
#include "../nodes/vp_sync_node.h" 
 
#include "../nodes/vp_screen_des_node.h" 
#include "../nodes/vp_rtsp_des_node.h" 
 
#include "../utils/analysis_board/vp_analysis_board.h" 
 
/* 
* ## firesmoke_detect_sample ## 
* detect firesmoke using yolo. 
*/ 
 
int main() { 
    VP_SET_LOG_LEVEL(vp_utils::vp_log_level::INFO); 
    VP_LOGGER_INIT(); 
 
    // create nodes 
    auto file_src_0 = std::make_shared<vp_nodes::vp_file_src_node>("file_src_0", 0, "./vp_data/test_video/smoke2.mp4", 0.5); 
    auto file_src_1 = std::make_shared<vp_nodes::vp_file_src_node>("file_src_1", 1, "./vp_data/test_video/fire.mp4", 0.5); 
    auto split0 = std::make_shared<vp_nodes::vp_split_node>("split0", false, true);  // split by deep-copy not by channel! 
    
    auto yolo_detector0 = std::make_shared<vp_nodes::vp_yolo_detector_node>("firesmoke_detector", "./vp_data/models/det_cls/firesmoke_yolov5s.onnx", "", "./vp_data/models/det_cls/firesmoke_3classes.txt", 640, 384); 
    auto yolo_detector1 = std::make_shared<vp_nodes::vp_yolo_detector_node>("yolo_detector", "./vp_data/models/det_cls/yolov3-tiny-2022-0721_best.weights", "./vp_data/models/det_cls/yolov3-tiny-2022-0721.cfg", "./vp_data/models/det_cls/yolov3_tiny_5classes.txt"); 
 
    auto osd = std::make_shared<vp_nodes::vp_osd_node>("osd"); 
    auto sync = std::make_shared<vp_nodes::vp_sync_node>("sync"); 
    auto screen_des_0 = std::make_shared<vp_nodes::vp_screen_des_node>("screen_des_0", 0);     
    auto screen_des_1 = std::make_shared<vp_nodes::vp_screen_des_node>("screen_des_1", 1); 
    auto split = std::make_shared<vp_nodes::vp_split_node>("split", true);  // split by deep-copy not by channel! 
 
    // construct pipeline 
    split0->attach_to({file_src_0,file_src_1}); 
    yolo_detector0->attach_to({split0}); 
    yolo_detector1->attach_to({split0}); 
    sync->attach_to({yolo_detector0, yolo_detector1}); 
    osd->attach_to({sync}); 
    split->attach_to({osd}); 
    screen_des_0->attach_to({split}); 
    screen_des_1->attach_to({split}); 
 
    file_src_0->start(); 
    file_src_1->start(); 
 
    // create nodes 
    auto file_src_01 = std::make_shared<vp_nodes::vp_file_src_node>("file_src_01", 2, "./vp_data/test_video/smoke2.mp4", 0.5); 
    auto file_src_11 = std::make_shared<vp_nodes::vp_file_src_node>("file_src_11", 3, "./vp_data/test_video/fire.mp4", 0.5); 
    auto split01 = std::make_shared<vp_nodes::vp_split_node>("split01", false, true);  // split by deep-copy not by channel! 
    
    auto yolo_detector01 = std::make_shared<vp_nodes::vp_yolo_detector_node>("firesmoke_detector1", "./vp_data/models/det_cls/firesmoke_yolov5s.onnx", "", "./vp_data/models/det_cls/firesmoke_3classes.txt", 640, 384); 
    auto yolo_detector11 = std::make_shared<vp_nodes::vp_yolo_detector_node>("yolo_detector1", "./vp_data/models/det_cls/yolov3-tiny-2022-0721_best.weights", "./vp_data/models/det_cls/yolov3-tiny-2022-0721.cfg", "./vp_data/models/det_cls/yolov3_tiny_5classes.txt"); 
 
    auto osd1 = std::make_shared<vp_nodes::vp_osd_node>("osd1"); 
    auto sync1 = std::make_shared<vp_nodes::vp_sync_node>("sync1"); 
    auto screen_des_01 = std::make_shared<vp_nodes::vp_screen_des_node>("screen_des_01", 2);     
    auto screen_des_11 = std::make_shared<vp_nodes::vp_screen_des_node>("screen_des_11", 3); 
    auto split1 = std::make_shared<vp_nodes::vp_split_node>("split1", true);  // split by deep-copy not by channel! 
 
    // construct pipeline 
    split01->attach_to({file_src_01,file_src_11}); 
    yolo_detector01->attach_to({split01}); 
    yolo_detector11->attach_to({split01}); 
    sync1->attach_to({yolo_detector01, yolo_detector11}); 
    osd1->attach_to({sync1}); 
    split1->attach_to({osd1}); 
    screen_des_01->attach_to({split1}); 
    screen_des_11->attach_to({split1}); 
 
    file_src_01->start(); 
    file_src_11->start(); 
 
    // for debug purpose 
    //vp_utils::vp_analysis_board board({file_src_0, file_src_1,file_src_01,file_src_11}); 
    //board.display(1, false); 
 
    std::string wait; 
    std::getline(std::cin, wait); 
    file_src_0->detach_recursively(); 
    file_src_1->detach_recursively(); 
    file_src_01->detach_recursively(); 
    file_src_11->detach_recursively(); 
}