Draws Keypoints

Draws Keypoints, an object describing a body part (like rightArm or leftShoulder), on given RGB tensor image.

draw_keypoints(
  image,
  keypoints,
  connectivity = NULL,
  colors = NULL,
  radius = 2,
  width = 3
)

Arguments

image

Tensor of shape (3 x H x W) and dtype uint8 or dtype float. In case of dtype float, values are assumed to be in range \([0, 1]\).

keypoints

Tensor of shape (N, K, 2) the K keypoints location for each of the N detected poses instance,

connectivity

List of integer pairs c(i, j) specifying which keypoints to connect with a line, e.g. list(c(1, 2), c(2, 3)). NULL (default) draws no connecting lines.

colors

character vector containing the colors of the keypoints or single color for all keypoints. The color can be represented as strings e.g. "red" or "#FF00FF". By default, rainbow colors are generated for keypoints

radius

radius of the plotted keypoint.

width

width of line connecting keypoints.

Value

Image Tensor of dtype uint8 with keypoints drawn.

See also

Other image display: draw_bounding_boxes(), draw_segmentation_masks(), tensor_image_browse(), tensor_image_display(), vision_make_grid()

Examples

if (torch::torch_is_installed()) {
if (FALSE) { # \dontrun{
image <- torch::torch_randint(190, 255, size = c(3, 360, 360))$to(torch::torch_uint8())
keypoints <- torch::torch_randint(low = 60, high = 300, size = c(4, 5, 2))
keypoint_image <- draw_keypoints(image, keypoints)
tensor_image_browse(keypoint_image)
} # }
}