如何增加pytorch张量的通道数

Question

我有一个格式为[N, 2, H, W]的pytorch张量，其中2是通道数。然而，我使用的模型(预训练的resnet18(要求我具有尺寸[N, 3, H, W]。如何将频道数量从2个增加到3个？

Answer 1

将2通道图像以灰度保存到磁盘，然后执行以下操作：

import torch
from PIL import Image
from torchvision.models import resnet18
from torchvision import transforms
transform = transforms.Compose([            
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize(
mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225]
)])
img= Image.open('pic.jpg').convert('RGB')
tensor= transform(img)
tensor= torch.unsqueeze(tensor, 0).float().cuda()
resnet_18_model= resnet18(pretrained= True).cuda() # resnet18()
resnet_18_model.eval()
output= resnet_18_model(tensor)
output= torch.argmax(output)
print('Class Number: ', output.item())