leon 1 месяц назад
Родитель
Сommit
165a2835db
2 измененных файлов с 3 добавлено и 4 удалено
  1. 0 1
      Makefile
  2. 3 3
      src/main.cpp

+ 0 - 1
Makefile

@@ -13,7 +13,6 @@ project_include_path := src
 opencv_include_path  := /workspace/compile/__install/opencv490/include/opencv4
 trt_include_path     := /usr/include/x86_64-linux-gnu/
 cuda_include_path    := $(cuda_home)/include
-cuvid_include_path   := /workspace/videoStreamInfer/src/cuvid-include
 ffmpeg_include_path  := 
 
 python_include_path  := /usr/include/python3.10

+ 3 - 3
src/main.cpp

@@ -24,13 +24,13 @@ int main()
     }; 
     std::shared_ptr<Infer> model = load("model/yolov8.engine", ModelType::YOLO11, names, 0, 0.5, 0.45);
 
-    std::shared_ptr<Node::StreamNode> src_node0   = std::make_shared<Node::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/101");
+    std::shared_ptr<Node::StreamNode> src_node0   = std::make_shared<Node::StreamNode>("src0", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/101", 0, DecodeType::GPU);
     src_node0->set_skip_frame(10);
 
-    std::shared_ptr<Node::StreamNode> src_node1   = std::make_shared<Node::StreamNode>("src1", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201");
+    std::shared_ptr<Node::StreamNode> src_node1   = std::make_shared<Node::StreamNode>("src1", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/201", 0, DecodeType::GPU);
     src_node1->set_skip_frame(10);
 
-    std::shared_ptr<Node::StreamNode> src_node2   = std::make_shared<Node::StreamNode>("src2", "rtsp://admin:lww123456@172.16.22.16:554/Streaming/Channels/301");
+    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, DecodeType::GPU);
     src_node2->set_skip_frame(10);