BoundingBoxes

class torchvision.tv_tensors.BoundingBoxes(data: Any, *, format: Union[BoundingBoxFormat, str], canvas_size: Tuple[int, int], dtype: Optional[dtype] = None, device: Optional[device, str, int]] = None, requires_grad: Optional[bool] = None)

torch.Tensor 子类，用于边界框，形状为 [N, 4]。

注意

每个样本应该只有一个 BoundingBoxes 实例，例如 {"img": img, "bbox": BoundingBoxes(...)}，尽管一个 BoundingBoxes 对象可以包含多个边界框。

参数:

• data – 任何可以使用 torch.as_tensor() 转换为张量的数据。

• format (BoundingBoxFormat, str) – 边界框的格式。

• canvas_size (python:int 的二元组) – 对应图像或视频的高度和宽度。

• dtype (torch.dpython:type, 可选) – 边界框的期望数据类型。如果省略，将从 data 推断。

• device (torch.device, 可选) – 边界框的期望设备。如果省略且 data 是 torch.Tensor，则设备将从中获取。否则，边界框将在 CPU 上构建。

• requires_grad (bool, 可选) – 自动求导是否应记录边界框上的操作。如果省略且 data 是 torch.Tensor，则值将从中获取。否则，默认为 False。

使用 BoundingBoxes 的示例

transforms v2 入门

transforms v2 入门

如何编写自己的 v2 transforms

如何编写自己的 v2 transforms

TVTensors 常见问题解答

TVTensors 常见问题解答

如何编写自己的 TVTensor 类

如何编写自己的 TVTensor 类