欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Pytorh 三通道的彩色图像转单通道的灰度图像

程序员文章站 2022-06-27 20:23:26
1. 三通道彩色图像转单通道灰度图像 做图像处理的宝宝们,通常会对输入的图像做处理,即图像增强。包括归一化,转tensor,旋转,翻转,裁剪,缩放等等操作,另外也可能会做三通道的彩色图像转单通道的灰度图像。 笔者看网上给了一些对源码操作的方式,我不推荐,一行代码就能搞定。transforms.Gra ......

1. 三通道彩色图像转单通道灰度图像

  做图像处理的宝宝们,通常会对输入的图像做处理,即图像增强。包括归一化,转tensor,旋转,翻转,裁剪,缩放等等操作,另外也可能会做三通道的彩色图像转单通道的灰度图像。

  笔者看网上给了一些对源码操作的方式,我不推荐,一行代码就能搞定。transforms.grayscale(num_output_channels=1)


2. 例子

2.1 code 

import torch
import torchvision
import torchvision.transforms as transforms
from pil import image


image_path = "/home/wuhf/whf/pytorch_code/dttnet/dataset/fusion_datasets/lytro-01-a.jpg"
image = image.open(image_path)
 
input_transform = transforms.compose([
   transforms.grayscale(1), #这一句就是转为单通道灰度图像
   transforms.totensor(),
])
image_tensor = input_transform(image)