sanitize_keypoints

torchvision.transforms.v2.functional.sanitize_keypoints(key_points: Tensor, canvas_size: Optional[tuple[int, int]] = None) → tuple[torch.Tensor, torch.Tensor]

移除图像区域外的关键点及其相应的标签（如果有）。

此变换会移除坐标超出其对应图像的“关键点”或“关键点组”及其相关的标签。如果您希望将此类关键点限制在图像边缘，请使用 ClampKeyPoints。

建议在流水线末尾调用此函数，然后再将输入传递给模型。

关键点可以作为一组单独的关键点传递，也可以作为一组对象（例如，多边形或多边形链）传递，这些对象由形状为 [..., 2] 的固定数量的关键点组成。当传递关键点组时（即，至少三维张量），此变换只会移除整个组，而不是组内的单个关键点。

参数:

• key_points (Tensor 或 KeyPoints) – 要清理的关键点。

• canvas_size (tuple of python:int, optional) – 关键点的画布大小（对应图像/视频的大小）。如果 key_points 是 KeyPoints 对象，则必须将其留空。

返回:

有效关键点的子集，以及相应的索引掩码。然后，该掩码可用于对与关键点关联的其他张量（例如标签）进行子集化。

返回类型:

out (tuple of Tensors)