RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimensio

使用transfroms方法过程报错：

RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0

代码块：

from PIL import Image
#from torchvision import transforms
#首先导入一张图片
img = Image.open("images/pytorch.png")
print(img)#运行一下看图片有没有正常读取
#它是PIL的图片类型，图片大小是955×500from torchvision import transforms
#首先看compose类，就是把不同的transforms结合在一起的#小课堂
#python中__call__的用法# transforms的compose方法既可以通过内置函数调用也可以通过__call__来调用
#常用一个类对象接一个括号()的形式调用#ToTensor
#将一个PIL或者numpy的图像转化成tensor
from torch.utils.tensorboard import SummaryWriter#从tensorboard引入了SummaryWriter
writer = SummaryWriter("logs")
img = Image.open("images/pytorch.png")
print(img)trans_totensor = transforms.ToTensor()
img_tensor = trans_totensor(img)
writer.add_image("ToTensor",img_tensor)
# writer.close()##Normalization归一化
# 能够归一化一个tnesor的图像，随着他的平均值或者标准差
# 给一个均值和标准差看维度
# 能够将每一个信道的值归一化# 归一化的计算公式
# output[channel] = (input[channel] - mean[channel]) / std[channel]`
print(img_tensor[0][0][0])
trans_norm = transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])
img_norm = trans_norm(img_tensor)
print(img_norm[0][0][0])
writer.add_image("normalize",img_norm)
writer.close()

报错后定位到第40行代码：

img_norm = trans_norm(img_tensor)

原因分析：

输入图像有一个 alpha 通道的像素，因此它是四通道而不是是三通道的图像

验证猜想

解决方法：

将四通道RGBA图像转换成三通道的RGB图像

img = Image.open("images/pytorch.png").convert('RGB')

RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimensio相关推荐

报错解决——RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton
问题描述 RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dim ...
The size of tensor a (4) must match the size of tensor b (3) at non-singletonThe size of
报错: RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton 解决: ...
The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0 维度不匹配
在导入一个png文件的时候,发现报这个错,源码如下,经过查询找到了两种解决方案. from PIL import Image from torchvision import transforms fr ...
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0—问题分析
声明: 这里仅仅提供一个思路,暂时没有绝对正确通用的解决方案作者:{ 墨理 } 感谢通过私信.评论交流,协力探讨解决问题的同学 UP .Penser 文章目录探讨的 pytorch 报错信息如下 ...
PyTorch 笔记（11）— Tensor内部存储结构（头信息区 Tensor，存储区 Storage）
1. Tensor 内部存储结构 tensor 数据结构如下图所示,tensor 分为头信息区(Tensor)和存储区 (Storage),信息区主要保存着 Tensor 的形状(size).步长(s ...
python中tensor与variable_NLP实战篇之tf2中tensor、variable、gradient、ops
本文是基于tensorflow2.2.0版本,介绍了tf中变量.张量的概念,tf中梯度的计算方式和tensor相关的操作. 实战系列篇章中主要会分享,解决实际问题时的过程.遇到的问题或者使用的工具等等 ...
Kernel size can‘t be greater than actual input size的解决方案
报错信息:RuntimeError: Calculated padded input size per channel: (49). Kernel size: (64). Kernel size ca ...
Pytorch踩坑: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.
报错信息: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 186 a ...
python size和count_groupby 的妙用（注意size和count）
Pandas的groupby()功能很强大,用好了可以方便的解决很多问题,在数据处理以及日常工作中经常能施展拳脚. 今天,我们一起来领略下groupby()的魅力吧. 首先,引入相关package: ...
entrypoint size limit: The following entrypoint(s) combined asset size exceeds the recommended limit
webpack打包提示文件体积过大导致 The following asset(s) exceed the recommended size limit (244 KiB). This can imp ...

RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimensio

使用transfroms方法过程报错：

代码块：

原因分析：

验证猜想

解决方法：

RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimensio相关推荐

最新文章

热门文章