边界框¶
- 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)[source]¶
torch.Tensor
的子类,表示形状为[N, 4]
的边界框。注意
每个样本应该只有一个
BoundingBoxes
实例,例如{"img": img, "bbox": BoundingBoxes(...)}
,尽管一个BoundingBoxes
对象可以包含多个边界框。- 参数:
data – 任何可以通过
torch.as_tensor()
转换为张量的数据。format (BoundingBoxFormat, str) – 边界框的格式。
canvas_size (two-tuple of python:ints) – 对应图像或视频的高度和宽度。
dtype (torch.dpython:type, optional) – 边界框所需的数据类型。如果省略,将从
data
推断。device (torch.device, optional) – 边界框所需的设备。如果省略且
data
是一个torch.Tensor
,则从data
获取设备。否则,边界框将在 CPU 上构建。requires_grad (bool, optional) – autograd 是否应记录在此边界框上的操作。如果省略且
data
是一个torch.Tensor
,则从data
获取该值。否则,默认为False
。
使用
BoundingBoxes
的示例