draw_keypoints

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

在给定的 RGB 图像上绘制关键点。图像值应为 uint8 类型的 [0, 255] 或浮点类型 [0, 1]。可以同时绘制多个实例的关键点。

此方法允许根据关键点的可见性来绘制关键点及其连接。

参数:

• image (Tensor) – 形状为 (3, H, W) 且 dtype 为 uint8 或 float 的张量。

• keypoints (Tensor) – 形状为 (num_instances, K, 2) 的张量，表示 N 个实例中每个实例的 K 个关键点位置，格式为 [x, y]。

• connectivity (List[Tuple[int, int]]]) – 一个元组列表，其中每个元组包含一对要连接的关键点。如果连接的两个关键点中至少有一个的 visibility 为 False，则不会绘制此特定连接。因不可见而排除的连接是按实例计算的。

• colors (str, Tuple) – 颜色可以表示为 PIL 字符串（例如，“red”或“#FF00FF”），或作为 RGB 元组（例如，(240, 10, 157)）。

• radius (int) – 表示关键点半径的整数。

• width (int) – 表示连接关键点的线的宽度的整数。

• visibility (Tensor) – 形状为 (num_instances, K) 的张量，指定 N 个实例中每个实例的 K 个关键点的可见性。True 表示相应的关键点可见并应被绘制。False 表示不可见，因此不会绘制该点或包含它的可能连接。输入张量将被转换为 bool 类型。默认值 None 表示所有关键点都可见。有关更多详细信息，请参阅 绘制带可见性的关键点。

返回:

绘制了关键点的图像张量。

返回类型:

img (Tensor[C, H, W])

使用 draw_keypoints 的示例

可视化工具

可视化工具