修改main

src/main.cpp

@@ -49,19 +49,19 @@ void test_yolo()
     std::shared_ptr<GNode::StreamNode> src_node0   = std::make_shared<GNode::StreamNode>("src0", "carperson.mp4", 1, GNode::DecodeType::GPU);
     src_node0->set_skip_frame(1);
 
-    std::shared_ptr<Infer> yolo_model = load("model/carperson.engine", ModelType::YOLOV5, names, 1, 0.25, 0.45);
-    std::shared_ptr<GNode::InferNode> infer_node   = std::make_shared<GNode::InferNode>("yolov5");
+    std::shared_ptr<Infer> yolo_model = load("model/carperson.engine", ModelType::YOLO11, names, 1, 0.25, 0.45);
+    std::shared_ptr<GNode::InferNode> infer_node   = std::make_shared<GNode::InferNode>("carperson_model");
     infer_node->set_model_instance(yolo_model, ModelType::YOLO11, 1);
 
     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::DrawNode> draw_node     = std::make_shared<GNode::DrawNode>("draw");
     std::shared_ptr<GNode::RecordNode> record_node = std::make_shared<GNode::RecordNode>("record");
     record_node->set_record_path("result/result.mp4");
     record_node->set_fps(25);
-    record_node->set_fourcc(cv::VideoWriter::fourcc('X', '2', '6', '4'));
+    record_node->set_fourcc(cv::VideoWriter::fourcc('m', 'p', '4', 'v'));
     
     GNode::LinkNode(src_node0, infer_node, max_size, stage);
     GNode::LinkNode(infer_node, track_node, max_size, stage);