How can I get the 'Test Picture for keypoints' picture?

[yeyuanzheng177]

[fastlater]

@yeyuanzheng177 any update. I only can see the training part, not the inference part. Maybe it is the visualize.draw_ai_boxes(image, refined_boxes=gt_bbox[:, :4], masks=gt_mask, bp=ai_class_names_)