|
@@ -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);
|
|
|
}
|
|
|
}
|
|
|
}
|