leon 1 lună în urmă
părinte
comite
77f6ebf2a5
2 a modificat fișierele cu 6 adăugiri și 3 ștergeri
  1. 4 1
      src/main.cpp
  2. 2 2
      src/nodes/stream/streamNode.cpp

+ 4 - 1
src/main.cpp

@@ -8,7 +8,10 @@
 
 void test_depth()
 {
-    std::shared_ptr<GNode::StreamNode> src_node0   = std::make_shared<GNode::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201", 0, GNode::DecodeType::GPU);
+    // std::shared_ptr<GNode::StreamNode> src_node0   = std::make_shared<GNode::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201", 0, GNode::DecodeType::GPU);
+    // src_node0->set_skip_frame(1);
+
+    std::shared_ptr<GNode::StreamNode> src_node0   = std::make_shared<GNode::StreamNode>("src0", "carperson.mp4", 0, GNode::DecodeType::GPU);
     src_node0->set_skip_frame(1);
 
     std::shared_ptr<Infer> depth_model = load("model/depth.engine", ModelType::DEPTH_ANYTHING, {}, 0, 0.25, 0.45);

+ 2 - 2
src/nodes/stream/streamNode.cpp

@@ -40,7 +40,7 @@ void StreamNode::work_cpu()
         }
 
         auto metaData = std::make_shared<meta::MetaData>();
-        metaData->image = frame;
+        metaData->image = frame.clone();
         metaData->from = name_;
 
         for (auto& output_buffer : output_buffers_)
@@ -82,7 +82,7 @@ void StreamNode::work_gpu()
             }
 
             auto metaData = std::make_shared<meta::MetaData>();
-            metaData->image = frame;
+            metaData->image = frame.clone();
             metaData->from = name_;
 
             for (auto& output_buffer : output_buffers_)