RGB

class torchvision.transforms.v2.RGB

将图像或视频转换为 RGB（如果它们本身不是 RGB）。

如果输入是 torch.Tensor，则期望其形状为 [..., 1 或 3, H, W]，其中 ... 表示任意数量的前导维度

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

用于覆盖自定义变换的方法。

请参阅 如何编写自己的 v2 变换