leon 3 周之前
父节点
当前提交
3692550703
共有 1 个文件被更改,包括 6 次插入7 次删除
  1. 6 7
      src/nodes/draw/drawNode.cpp

+ 6 - 7
src/nodes/draw/drawNode.cpp

@@ -137,7 +137,6 @@ void DrawNode::handle_data(std::shared_ptr<meta::MetaData>& meta_data)
         {
             for (const auto& point : box.keypoints)
             {
-                std::tie(b, g, r) = random_color(box.class_id);
                 cv::circle(image, cv::Point(point.x, point.y), 5, cv::Scalar(b, g, r), -1);
             }
             for (const auto& pair : coco_pairs) 
@@ -145,13 +144,13 @@ void DrawNode::handle_data(std::shared_ptr<meta::MetaData>& meta_data)
                 int startIdx = pair.first;
                 int endIdx = pair.second;
 
-                if (startIdx < obj.keypoints.size() && endIdx < obj.keypoints.size()) 
+                if (startIdx < box.keypoints.size() && endIdx < box.keypoints.size()) 
                 {
-                    int x1 = (int)obj.keypoints[startIdx].x;
-                    int y1 = (int)obj.keypoints[startIdx].y;
-                    int x2 = (int)obj.keypoints[endIdx].x;
-                    int y2 = (int)obj.keypoints[endIdx].y;
-                    cv::line(image, cv::Point(x1, y1), cv::Point(x2, y2), cv::Scalar(255, 0, 0), 2);
+                    int x1 = (int)box.keypoints[startIdx].x;
+                    int y1 = (int)box.keypoints[startIdx].y;
+                    int x2 = (int)box.keypoints[endIdx].x;
+                    int y2 = (int)box.keypoints[endIdx].y;
+                    cv::line(image, cv::Point(x1, y1), cv::Point(x2, y2), cv::Scalar(b, g, r), 2);
                 }
             }
         }