绘制关键点¶
- torchvision.utils.draw_keypoints(image: Tensor, keypoints: Tensor, connectivity: Optional[List[Tuple[int, int]]] = None, colors: Optional[Union[str, Tuple[int, int, int]]] = None, radius: int = 2, width: int = 3, visibility: Optional[Tensor] = None) Tensor[source]¶
在给定的RGB图像上绘制关键点。 图像值应为[0, 255]范围内的uint8类型或[0, 1]范围内的浮点类型。 可以同时为多个实例绘制关键点。
此方法允许根据关键点的可见性绘制关键点及其连接性。
- Parameters:
image (Tensor) – 形状为 (3, H, W) 且数据类型为 uint8 或 float 的张量。
关键点 (张量) – 形状为 (num_instances, K, 2) 的张量，表示每个 N 实例的 K 个关键点位置，格式为 [x, y]。
连接性 (列表[元组[整数, 整数]]]) – 一个元组列表，其中每个元组包含一对要连接的关键点。 如果两个连接的关键点中至少有一个的
可见性为False， 则不会绘制此特定连接。 由于不可见性导致的排除是按实例计算的。颜色 (str, Tuple) – 颜色可以表示为PIL字符串，例如“红色”或“#FF00FF”，或者作为RGB元组，例如
(240, 10, 157)。radius (int) – 表示关键点半径的整数。
width (int) – 表示连接关键点的线宽度的整数。
visibility (Tensor) – 形状为 (num_instances, K) 的张量，指定每个 N 实例的 K 个关键点的可见性。 True 表示相应的关键点是可见的并且应该被绘制。 False 表示不可见，因此不会绘制该点以及包含它的可能连接。 输入张量将被转换为布尔类型。 默认
None表示所有关键点都是可见的。 更多详情，请参见 绘制带有可见性的关键点。
- Returns:
绘制了关键点的图像张量。
- Return type:
img (张量[C, H, W])
使用
draw_keypoints的示例：
