【深度学习】【预处理】pytorch输入图片单通道转多通道
大部分的模型在输入端要求图像为三通道,单通道的图像输入一般会报错,
在pytorch的Transform中可实现三通道与单通道的互转
第一种方式:
tf=Transform.Compose([
lambda x:Image.open(x).convert('RGB'),#此处为单通道转RGB,将RGB换为L可转为单通道
transform.toTensor()]
第二种方式
tf=Transform.Compose([
transforms.Grayscale(num_output_channels=3),#3为三通道,1为单通道
transforms.toTensor()]
注:单通道转多通道的原理实际上是将单通道的图像复制3份,在显示结果上不存在差异(Mean与Std的通道数量需要相应调整)