draw_keypoints: Draws Keypoints

In torchvision: Models, Datasets and Transformations for Images

Draws Keypoints

Description

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

Usage

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

Arguments

image	: Tensor of shape (3, H, W) and dtype uint8
keypoints	: Tensor of shape (N, K, 2) the K keypoints location for each of the N detected poses instance,
connectivity	: Vector of pair of keypoints to be connected (currently unavailable)
colors	: character vector containing the colors of the boxes or single color for all boxes. The color can be represented as strings e.g. "red" or "#FF00FF". By default, viridis 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.

Examples

if (torch::torch_is_installed()) {
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)
}

torchvision documentation built on April 14, 2023, 5:08 p.m.