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:ints 的二元組) – 對應圖像或影片的高度和寬度。

• dtype (torch.dpython:type, optional) – 邊界框的所需資料類型。如果省略，將從 data 推斷。

• device (torch.device, optional) – 邊界框的所需裝置。如果省略且 data 是一個 torch.Tensor，則從中獲取裝置。否則，邊界框將在 CPU 上構建。

• requires_grad (bool, optional) – 是否 autograd 應該記錄邊界框上的操作。 如果省略且 data 是一個 torch.Tensor，則從中獲取值。 否則，預設為 False。

使用 BoundingBoxes 的範例

Transforms v2 入門

Transforms v2 入門

如何編寫您自己的 v2 轉換

如何編寫您自己的 v2 轉換

TVTensors 常見問題

TVTensors 常見問題

如何編寫您自己的 TVTensor 類別

如何編寫您自己的 TVTensor 類別