BoundingBoxes

class torchvision.tv_tensors.BoundingBoxes(data: Any, *, format: Union[BoundingBoxFormat, str], canvas_size: Tuple[int, int], dtype: Optional[dtype] = None, device: Optional[Union[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 的示例

变换 v2 入门

变换 v2 入门

如何编写自己的 v2 变换

如何编写自己的 v2 变换

TVTensor 常见问题

TVTensor 常见问题

如何编写自己的 TVTensor 类

如何编写自己的 TVTensor 类