test

src/main.cpp

@@ -3,6 +3,7 @@
 #include "nodes/infer/inferNode.hpp"
 #include "nodes/draw/drawNode.hpp"
 #include "nodes/track/trackNode.hpp"
+#include "nodes/analyze/analyzeNode.hpp"
 #include "nodes/httpPush/httpPush.hpp"
 #include "nodes/record/recordNode.hpp"
 
@@ -46,6 +47,8 @@ void test_yolo()
 
     std::shared_ptr<GNode::TrackNode> track_node     = std::make_shared<GNode::TrackNode>("tracker", "person", 30, 30);
 
+    std::shared_ptr<GNode::AnalyzeNode> analyze_node = std::make_shared<GNode::AnalyzeNode>("analyze");
+
     std::shared_ptr<GNode::DrawNode> draw_node     = std::make_shared<GNode::DrawNode>("draw_track");
     std::shared_ptr<GNode::RecordNode> record_node = std::make_shared<GNode::RecordNode>("record");
     record_node->set_record_path("result/result.mp4");
@@ -54,10 +57,12 @@ void test_yolo()
     
     GNode::LinkNode(src_node0, infer_node, max_size, stage);
     GNode::LinkNode(infer_node, track_node, max_size, stage);
-    GNode::LinkNode(track_node, draw_node, max_size, stage);
+    GNode::LinkNode(track_node, analyze_node, max_size, stage);
+    GNode::LinkNode(analyze_node, draw_node, max_size, stage);
     GNode::LinkNode(draw_node, record_node, max_size, stage);
     record_node->start();
     draw_node->start();
+    analyze_node->start();
     track_node->start();
     infer_node->start();
     src_node0->start();