RGB

class torchvision.transforms.v2.RGB

將影像或影片轉換為 RGB（如果它們不是 RGB）。

如果輸入是 torch.Tensor，則其形狀預計為 [..., 1 或 3, H, W]，其中 ... 表示任意數量的前導維度

transform(inpt: Any, params: dict[str, Any]) → Any

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

請參閱 如何編寫自己的 v2 變換