torchvision.transforms 灰度图
时间: 2023-03-23 20:01:29 浏览: 355
灰度图的转换
3星 · 编辑精心推荐
torchvision.transforms中的Grayscale()函数可以将彩色图像转换为灰度图像。该函数接受一个参数num_output_channels,用于指定输出图像的通道数,默认值为1,即输出单通道灰度图像。如果将num_output_channels设置为3,则输出3通道灰度图像,其中每个通道都具有相同的值,即相同的灰度值。
阅读全文