4、图片转为灰度图 transforms.Grayscale()

啥都亿点点的研究生

已于 2022-04-01 09:08:41 修改

阅读量6.8k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： pytorch数据增强方法文章标签：计算机视觉 python opencv

于 2022-04-01 08:57:57 首次发布

本文链接：https://blog.youkuaiyun.com/lichaoqi1/article/details/123889076

pytorch数据增强方法专栏收录该内容

24 篇文章

订阅专栏

该博客演示了如何使用 PyTorch 的 torchvision 库将彩色图像转换为灰度图像。通过导入必要的库，加载一张图片，然后应用 torchvision.transforms.Grayscale() 函数，可以看到原始图像与转换后的灰度图像的对比。这有助于理解在计算机视觉任务中图像预处理的重要性。

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

 transforms.Grayscale()
num_output_channels=1 or 3

import torchvision.transforms as transform
from PIL import Image
import matplotlib.pyplot as plt
import numpy as np
import torch
img0=Image.open('lin-xiao-xun-000003.jpg')
img1=transform.Grayscale(3)(img0)
axs = plt.figure().subplots(1, 2)
axs[0].imshow(img0);axs[0].set_title('src');axs[0].axis('off')
axs[1].imshow(img1,cmap=plt.get_cmap('gray'));axs[1].set_title('gray');axs[1].axis('off')
plt.show()