stack expects each tensor to be equal size, but got [3, 224, 224] at entry 0 and [1,224,224] at entr

首先可以确定地是在数据读取的以及处理发生的问题。目前可以归纳为一下两种：

1：使用Image.open（）没有添加.convert('RGB')

错误代码例子：

class Mydataset(data.Dataset):def __init__(self,root,labels,transform):super(Mydataset,self).__init__()self.imgs_path = rootself.labels = labelsself.transform = transformdef __getitem__(self,index):ig_path = self.imgs_path[index]label=self.labels[index]######################################pil_image = Image.open(ig_path)#########################################data = self.transform(pil_image)return data,labeldef __len__(self):return len(self.imgs_path)

正确代码例子：

class Mydataset(data.Dataset):def __init__(self,root,labels,transform):super(Mydataset,self).__init__()self.imgs_path = rootself.labels = labelsself.transform = transformdef __getitem__(self,index):ig_path = self.imgs_path[index]label=self.labels[index]######################################pil_image = Image.open(ig_path).convert('RGB')#########################################data = self.transform(pil_image)return data,labeldef __len__(self):return len(self.imgs_path)

2：transform过程中错误使用Resize功能

错误示范：

transform = transforms.Compose([transforms.Resize((224)),transforms.ToTensor()
])

正确代码：

train_dataset = datasets.ImageFolder(train_data,transforms.Compose([transforms.Resize((224，224))

验证方法：

path_load = glob.glob(r'D:\BaiduNetdiskDownload\pytorch_learning\dataset\dataset2\*.jpg')
#图片路径
transform = transforms.Compose([transforms.Resize((224,224)),transforms.ToTensor()
])all_labels=[]
species = ['cloudy', 'rain', 'shine', 'sunrise']
#图片类别
for img in path_load:for i,c in enumerate(species):if c in img:all_labels.append(i)
#图片标签
species_to_idx = dict((c, i) for i, c in enumerate(species))
label_to_class =  dict((v,k) for k,v in species_to_idx.items())
class Mydataset(data.Dataset):def __init__(self,root,labels,transform):super(Mydataset,self).__init__()self.imgs_path = rootself.labels = labelsself.transform = transformdef __getitem__(self,index):ig_path = self.imgs_path[index]label=self.labels[index]pil_image = Image.open(ig_path).convert('RGB')data = self.transform(pil_image)return data,labeldef __len__(self):return len(self.imgs_path)wheather_dataset = Mydataset(path_load,all_labels,transform)
wheather_dl = data.DataLoader(wheather_dataset,batch_size=16,shuffle=True,drop_last=True)plt.figure(figsize=(12,8))
imgs_batch,labels_batch=next(iter(wheather_dl))
for i,(img,label) in enumerate(zip(imgs_batch,labels_batch)):img = img.permute(1,2,0).numpy()plt.subplot(4,4,i+1)plt.title(label_to_class.get(label.item()))plt.imshow(img)

result：

stack expects each tensor to be equal size, but got [3, 224, 224] at entry 0 and [1,224,224] at entr相关推荐

stack expects each tensor to be equal size, but got [3, 40, 160] at entry 0 and [4, 40, 160] at entr
这个问题是通道数不一样导致的,本质上是图像的问题加载图像时加上.convert('RGB')即可 data = Image.open(imgPath).convert('RGB')
DataLoader问题解决：RuntimeError: stack expects each tensor to be equal size, but got [3, 200, 200]entry1
最近,在数据集处理并载入DataLoader进行训练的时候出现了问题: RuntimeError: stack expects each tensor to be equal size, but go ...
RuntimeError: stack expects each tensor to be equal size, but got xxx at entry 0 at entry 1
今日做模型训练,Pytorch在加载数据时遇到如下错误: Epoch 1/800: 31%|███ | 4/13 [00:03<00:07, 1.27img/s, loss (batch)=1. ...
RuntimeError: stack expects each tensor to be equal size
在调试densenet进行分类任务的代码时,在图像预处理的过程中遇到下列错误: RuntimeError: stack expects each tensor to be equal size, bu ...
RuntimeError: stack expects each tensor to be equal size, but got [8] at entry 0 and [2] at entry 2
最近在调试pytorch代码的时候遇到如下问题,由于他报错的地方不是在我们自己写的代码,而是在pytorch的包里,所以一开始就一头雾水. 在查阅了资料以后http://www.zzvips. ...
RuntimeError: each element in list of batch should be of equal size
RuntimeError: each element in list of batch should be of equal size 示例代码: import os import re from t ...
解决RuntimeError: stack expects a non-empty Tensorlist问题
在batch_size时,可能某些地方需要batch>1的时候才能运行,但在创建dataloader时,可能最后一个batch的大小正好为1,所以会导致这种错误. 为了解决这种情况,将datal ...
Cityscapse 数据集使用 + 训练 STDC踩坑
官网地址:Cityscapes Dataset – Semantic Understanding of Urban Street Scenes (cityscapes-dataset.com) get ...
初学者之蚂蚁蜜蜂分类报错记录
听了bilibil小土堆的课,拿蚂蚁蜜蜂数据集练手,记录下自己犯的错~ 1.trans = transform.Compose() # 这段报错了-- #图片转为tensor 修改维度 trans = ...

stack expects each tensor to be equal size, but got [3, 224, 224] at entry 0 and [1,224,224] at entr

验证方法：

stack expects each tensor to be equal size, but got [3, 224, 224] at entry 0 and [1,224,224] at entr相关推荐

最新文章

热门文章