bugfix

src/nodes/track/trackNode.cpp

@@ -6,10 +6,6 @@ namespace GNode
 void TrackNode::work()
 {
     printf("TrackNode %s\n", name_.c_str());
-    for (const auto& input_buffer : input_buffers_)
-    {
-        tracker_[input_buffer.first] = std::make_shared<BYTETracker>(30, 30);
-    }
     while (running_)
     {
         bool has_data = false;
@@ -24,7 +20,7 @@ void TrackNode::work()
             // printf("Node %s get data from %s\n", name_.c_str(), input_buffer.first.c_str());
 
             // auto res = model_->forward(tensor::cvimg(image), image.cols, image.rows, 0.0f, 0.0f);
-            if (!tracker_[input_buffer.first])
+            if (!tracker_)
             {
                 printf("track is nullptr\n");
                 continue;
@@ -42,7 +38,7 @@ void TrackNode::work()
                     return obj;
                 }
             });
-            std::vector<STrack> output_stracks = tracker_[input_buffers.first]->update(objects);
+            std::vector<STrack> output_stracks = tracker_->update(objects);
             if (output_stracks.size() > objects.size())
             {
                 printf("output_stracks size: %d, objects size: %d\n", output_stracks.size(), objects.size());

src/nodes/track/trackNode.hpp

@@ -18,13 +18,13 @@ public:
     TrackNode(const std::string& name, const std::string& track_label, int frame_rate, int track_buffer) : BaseNode(name, NODE_TYPE::MID_NODE) 
     {
         track_label_ = track_label;
-        // tracker_ = std::make_shared<BYTETracker>(frame_rate, track_buffer);
+        tracker_ = std::make_shared<BYTETracker>(frame_rate, track_buffer);
     }
     virtual ~TrackNode()  { stop(); };
     void work() override;
 
 private:
-    std::unordered_map<std::string, std::shared_ptr<BYTETracker>> tracker_;
+    std::shared_ptr<BYTETracker> tracker_;
     std::string track_label_;
 
 };