快捷方式

RandomGrayscale

class torchvision.transforms.v2.RandomGrayscale(p: float = 0.1)[source]

以概率 p(默认为 0.1)随机将图像或视频转换为灰度。

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

输出与输入具有相同数量的通道。

参数:

p (float) – 图像应转换为灰度的概率。

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

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

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

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

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

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源