leon 3 周之前
父節點
當前提交
37566b834d
共有 2 個文件被更改,包括 4 次插入1 次删除
  1. 2 0
      src/nodes/infer/inferNode.cpp
  2. 2 1
      src/nodes/track/trackNode.cpp

+ 2 - 0
src/nodes/infer/inferNode.cpp

@@ -37,6 +37,8 @@ void InferNode::handle_data(std::shared_ptr<meta::MetaData>& meta_data)
         auto result = std::get<data::BoxArray>(det_result);
         for (auto& r : result)
         {
+            // 默认将id设置为-1,复用目标检测的id作为追踪的id, 只有经过追踪的目标才会设置id
+            r.class_id = -1;
             meta_data->boxes.push_back(r);
         }
     } 

+ 2 - 1
src/nodes/track/trackNode.cpp

@@ -19,7 +19,8 @@ void TrackNode::handle_data(std::shared_ptr<meta::MetaData>& meta_data)
             obj.rect.y = box.top;
             obj.rect.width = box.right - box.left;
             obj.rect.height = box.bottom - box.top;
-            obj.label = box.class_id; // 假设 Object::label 存的是 int 类型的 class_id
+            // obj.label = box.class_id; // 假设 Object::label 存的是 int 类型的 class_id
+            obj.label = 0;
             obj.prob = box.score;
 
             if (obj.rect.width > 0 && obj.rect.height > 0 && obj.prob > 0) { // 至少prob > 0