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.uj5u.com/houduan/150257.html
標籤：Python