Pytorh 三通道的彩色圖像轉單通道的灰度圖像

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)