Explorar o código

test yolo depth

leon hai 5 meses
pai
achega
1d76e43971
Modificáronse 1 ficheiros con 8 adicións e 8 borrados
  1. 8 8
      src/main.cpp

+ 8 - 8
src/main.cpp

@@ -63,9 +63,9 @@ void test_multi()
     std::shared_ptr<Node::StreamNode> src_node2   = std::make_shared<Node::StreamNode>("src2", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/301", 0, Node::DecodeType::GPU);
     src_node2->set_skip_frame(10);
 
-    // std::shared_ptr<Infer> yolo_model = load("model/yolo11s.engine", ModelType::YOLO11, names, 0, 0.25, 0.45);
-    // std::shared_ptr<Node::InferNode> infer_node1   = std::make_shared<Node::InferNode>("yolo11");
-    // infer_node1->set_model_instance(yolo_model, ModelType::YOLO11);
+    std::shared_ptr<Infer> yolo_model = load("model/yolo11s.engine", ModelType::YOLO11, names, 0, 0.25, 0.45);
+    std::shared_ptr<Node::InferNode> infer_node1   = std::make_shared<Node::InferNode>("yolo11");
+    infer_node1->set_model_instance(yolo_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");
@@ -73,11 +73,11 @@ void test_multi()
 
     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_node2);
-    Node::LinkNode(src_node1, infer_node2);
-    Node::LinkNode(src_node2, infer_node2);
-    // Node::LinkNode(infer_node1, infer_node2);
+
+    Node::LinkNode(src_node0, infer_node1);
+    Node::LinkNode(src_node1, infer_node1);
+    Node::LinkNode(src_node2, infer_node1);
+    Node::LinkNode(infer_node1, infer_node2);
     Node::LinkNode(infer_node2, draw_node);
     Node::LinkNode(draw_node, push_node);
     push_node->start();