邊界框

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 (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 的示例

如何編寫您自己的 v2 變換

如何編寫您自己的 v2 變換

如何編寫您自己的 TVTensor 類

如何編寫您自己的 TVTensor 類

v2 變換入門

v2 變換入門

TVTensors 常見問題解答

TVTensors 常見問題解答