我有一些带有标签的灰度和彩色图像。我想将这种灰色和彩色图像（4通道）结合起来，并使用4通道图像运行转移学习。怎么做？
如果我正确理解了这个问题，则希望将1通道图像和3通道图像组合在一起，并获得4通道图像并将其用作输入。
如果这是您要执行的操作，则可以使用torch.cat（）。
加载两个图像并将其沿通道尺寸合并的一些示例代码
import numpy as np
import torch
from PIL import Image
image_rgb = Image.open(path_to_rgb_image)
image_rgb_tensor = torch.from_numpy(np.array(image_rgb))
image_rgb.close()
image_grayscale = Image.open(path_to_grayscale_image))
image_grayscale_tensor = troch.from_numpy(np.array(image_grayscale))
image_grayscale.close()
image_input = torch.cat([image_rgb_tensor, image_grayscale_tensor], dim=2)
我假设您要使用的灰度图像转换为具有形状的张量，[..., ..., 1]而rgb图像转换为[..., ..., 3]。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句