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)