Pytorch——TypeError：无法处理此数据类型

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")