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

2024年6月9日 244次阅读 来源: 抚琴尘世客

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)

 

    原文作者：抚琴尘世客
    原文地址: https://www.cnblogs.com/haifwu/p/12825511.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞