from torchvision import transforms



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

 

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐