pytorch彩色图像转灰度图像

from torchvision import transforms



transform = transforms.Compose([
    transforms.Grayscale(num_output_channels=1), # 彩色图像转灰度图像num_output_channels默认1
    transforms.ToTensor()
])

 

你可能感兴趣的:(小技巧)