快捷方式

随机灰度化

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

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

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

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

参数::

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

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

查看资源