SanitizeKeyPoints

class torchvision.transforms.v2.SanitizeKeyPoints(labels_getter: Optional[Union[Callable[[Any], Any], str]] = None)

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

此转换会移除坐标超出相应图像范围的关键点或关键点组及其关联的标签。如果您希望将此类关键点钳制到图像边缘，请使用 ClampKeyPoints。

建议在管道的最后调用它，然后再将输入传递给模型。

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

参数:

labels_getter (callable 或 str 或 None, optional) –

指示如何识别输入中的标签（或需要与关键点一起清理的任何其他内容）。如果设置为字符串 "default"，它将尝试在输入中查找“labels”键（不区分大小写），如果输入是字典，或者是一个元组，其第二个元素是字典。

它也可以是一个可调用对象，该对象接受与转换相同的输入，并返回以下之一：

• 单个张量（标签）

• 张量元组/列表，每个张量都将像关键点一样被清理。

如果 labels_getter 为 None（默认值），则只清理关键点。

forward(*inputs: Any) → Any

不要覆盖此方法！请使用 transform() 代替。

transform(inpt: Any, params: dict[str, Any]) → Any

用于覆盖自定义变换的方法。

请参阅 如何编写自己的 v2 变换