Browse Source

test depth

leon 1 month ago
parent
commit
d300f81865
1 changed files with 9 additions and 3 deletions
  1. 9 3
      src/main.cpp

+ 9 - 3
src/main.cpp

@@ -19,19 +19,25 @@ int main()
     src_node2->set_skip_frame(10);
 
     std::shared_ptr<Infer> model = load("model/yolo11s.engine", ModelType::YOLO11, names, 0, 0.25, 0.45);
-    std::shared_ptr<Node::InferNode> infer_node   = std::make_shared<Node::InferNode>("infer");
+    std::shared_ptr<Node::InferNode> infer_node1   = std::make_shared<Node::InferNode>("YOLO11");
     infer_node->set_model_instance(model, ModelType::YOLO11);
 
+    std::shared_ptr<Infer> depth_model = load("model/depth.engine", ModelType::DEPTH_ANYTHING, {}, 0, 0.25, 0.45);
+    std::shared_ptr<Node::InferNode> infer_node2   = std::make_shared<Node::InferNode>("DEPTH_ANYTHING");
+    infer_node2->set_model_instance(depth_model, ModelType::DEPTH_ANYTHING);
+
     std::shared_ptr<Node::DrawNode> draw_node     = std::make_shared<Node::DrawNode>("draw");
     std::shared_ptr<Node::HttpPushNode> push_node = std::make_shared<Node::HttpPushNode>("push", "172.16.20.168", 8080, "/push");
     Node::LinkNode(src_node0, infer_node);
     Node::LinkNode(src_node1, infer_node);
     Node::LinkNode(src_node2, infer_node);
-    Node::LinkNode(infer_node, draw_node);
+    Node::LinkNode(infer_node1, infer_node2);
+    Node::LinkNode(infer_node2, draw_node);
     Node::LinkNode(draw_node, push_node);
     push_node->start();
     draw_node->start();
-    infer_node->start();
+    infer_node1->start();
+    infer_node2->start();
     src_node0->start();
     src_node1->start();
     src_node2->start();