当前位置： 移动技术网 > IT编程>脚本编程>Python > Pytorh 三通道的彩色图像转单通道的灰度图像

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

2020年05月04日  | 移动技术网IT编程  | 我要评论

天津警备区,火鸟,条码打印机设备

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)

 

您可能感兴趣的文章:

如对本文有疑问，请在下面进行留言讨论，广大热心网友会与你互动！！ 点击进行留言回复

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网