快捷方式

RandomPerspective

class torchvision.transforms.v2.RandomPerspective(distortion_scale: float = 0.5, p: float = 0.5, interpolation: Union[InterpolationMode, int] =InterpolationMode.BILINEAR, fill: Union[int, float, Sequence[int], Sequence[float], None, Dict[Union[Type, str], Optional[Union[int, float, Sequence[int], Sequence[float]]]]] = 0)[source]

以給定的機率對輸入執行隨機透視變換。

如果輸入是 torch.TensorTVTensor(例如 ImageVideoBoundingBoxes 等),它可以具有任意數量的前導批次維度。例如,影像可以具有 [..., C, H, W] 的形狀。邊界框可以具有 [..., 4] 的形狀。

引數:
  • distortion_scale (float, 可選) – 控制失真程度的引數,範圍從 0 到 1。預設值為 0.5。

  • p (float, 可選) – 輸入被轉換的機率。預設值為 0.5。

  • interpolation (InterpolationMode, 可選) – 由 torchvision.transforms.InterpolationMode 定義的所需插值列舉。預設值為 InterpolationMode.BILINEAR。如果輸入是 Tensor,僅支援 InterpolationMode.NEARESTInterpolationMode.BILINEAR。相應的 Pillow 整型常量,例如 PIL.Image.BILINEAR 也被接受。

  • fill (數字tupledict, 可選) – 當 padding_mode 為 constant 時使用的畫素填充值。預設值為 0。如果是一個長度為 3 的 tuple,則分別用於填充 R、G、B 通道。填充值也可以是一個將資料型別對映到填充值的字典,例如 fill={tv_tensors.Image: 127, tv_tensors.Mask: 0},其中 Image 將填充 127,Mask 將填充 0。

使用 RandomPerspective 的示例

變換示例

變換示例
static get_params(width: int, height: int, distortion_scale: float) Tuple[List[List[int]], List[List[int]]][source]

獲取用於隨機透視變換的 perspective 引數。

引數:
  • width (int) – 影像寬度。

  • height (int) – 影像高度。

  • distortion_scale (float) – 控制失真程度的引數,範圍從 0 到 1。

返回值:

包含原始影像 [左上、右上、右下、左下] 座標的列表,包含變換後圖像 [左上、右上、右下、左下] 座標的列表。

make_params(flat_inputs: List[Any]) Dict[str, Any][source]

用於自定義變換的覆蓋方法。

參見 如何編寫您自己的 v2 變換

transform(inpt: Any, params: Dict[str, Any]) Any[source]

用於自定義變換的覆蓋方法。

參見 如何編寫您自己的 v2 變換

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源