python - Pytorch——TypeError: Cannot handle this data type

Question

我正在做一个图像处理任务，我想使用 torch.cat 来连接属于两个不同文件夹的图片。文件夹1的图片大小为224*224*3，文件夹2的图片大小为224*224*1。之后，我想保存生成的图片。我使用了保存功能，但是我遇到错误，错误代码是“TypeError：无法处理此数据类型”。请帮我找到解决方案。这是我的代码：

import glob
import os
import torch
import torchvision.transforms.functional as TF
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
import scipy.misc

f1_folder = 'F:\\picture1'
f2_folder = 'F:\\picture2'

f1_images = glob.glob(os.path.join(f1_folder, '*.jpg'))
f2_images = glob.glob(os.path.join(f2_folder, '*.jpg'))

for f1_img, f2_img in zip(f1_images, f2_images):

img1 = Image.open(f1_img)
img2 = Image.open(f2_img)

cat_img = torch.cat((TF.to_tensor(img1), TF.to_tensor(img2)), dim=1)# for 'CHW'
cat_img = cat_img.numpy()  
cat_img = np.transpose(cat_img, (1, 2, 0))  

im = Image.fromarray(cat_img)
im.save("file.jpeg")

标签： python