pytorch彩色图像转灰度图像

嘿嘿哈嘻

于 2020-06-27 16:03:49 发布

阅读量7.2k

点赞数 3

CC 4.0 BY-SA版权

分类专栏：小技巧

本文链接：https://blog.youkuaiyun.com/adaadadsa/article/details/106983918

小技巧专栏收录该内容

62 篇文章

订阅专栏

本文详细介绍使用PyTorch的transforms模块将彩色图像转换为灰度图像并进行张量化的过程。通过实例演示如何利用Compose组合多个转换操作，实现图像预处理自动化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from torchvision import transforms



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