leon 1 kuukausi sitten
vanhempi
commit
8541c90cac
2 muutettua tiedostoa jossa 3 lisäystä ja 2 poistoa
  1. 2 1
      src/main.cpp
  2. 1 1
      src/nodes/track/trackNode.hpp

+ 2 - 1
src/main.cpp

@@ -39,7 +39,8 @@ void test_yolo()
     std::shared_ptr<GNode::InferNode> infer_node   = std::make_shared<GNode::InferNode>("yolov5");
     infer_node->set_model_instance(yolo_model, ModelType::YOLO11);
 
-    std::shared_ptr<GNode::TrackNode> track_node     = std::make_shared<GNode::TrackNode>("tracker", {"person"}, 30, 30);
+    std::vector<std::string> tracker_label = { "person"};
+    std::shared_ptr<GNode::TrackNode> track_node     = std::make_shared<GNode::TrackNode>("tracker", tracker_label, 30, 30);
 
     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");

+ 1 - 1
src/nodes/track/trackNode.hpp

@@ -15,7 +15,7 @@ class TrackNode : public BaseNode
 {
 public:
     TrackNode() = delete;
-    TrackNode(const std::string& name, std::vector<std::string> track_labels, int frame_rate, int track_buffer) : BaseNode(name, NODE_TYPE::MID_NODE) 
+    TrackNode(const std::string& name, const std::vector<std::string>& track_labels, int frame_rate, int track_buffer) : BaseNode(name, NODE_TYPE::MID_NODE) 
     {
         track_labels_ = track_labels;
         tracker_ = std::make_shared<BYTETracker>(frame_rate, track_buffer);