PyTorch 解决Dataset和Dataloader遇到的问题
更新时间：2020年01月08日 14:14:34 作者：xgbm_k
今天小编就为大家分享一篇PyTorch 解决Dataset和Dataloader遇到的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
今天在使用PyTorch中Dataset遇到了一个问题。先看代码
class psDataset(Dataset):
def __init__(self, x, y, transforms = None):
super(Dataset, self).__init__()
self.x = x
self.y = y
if transforms == None:
self.transforms = Compose([Resize((224, 224)), ToTensor()])
else:
self.transforms = transforms
def __len__(self):
return len(self.x)
def __getitem__(self, idx):
img = Image.open(self.x[idx])
img = self.transforms(img)
return img, torch.tensor([[self.y[idx]]])
结果运行时报错：RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 1 in dimension 1 at /opt/conda/conda-bld/pytorch_1522182087074/work/torch/lib/TH/generic/THTensorMath.c:2897
Google了一下发现是这样的：读入的图片有些是灰度图（1个通道），绝大多数是RGB图片（3通道），也有些是带透明度的（4通道）
。这导致在读入后最后一个维度（通道数）不一致（可能是1、3或者4）。
Dataloader在制作batch data时，tensor的shape必须一样，就报了这个错误。解决的方法是：img = img.convert(“RGB”)。完
整代码如下：
class psDataset(Dataset):
def __init__(self, x, y, transforms = None):
super(Dataset, self).__init__()
self.x = x
self.y = y
if transforms == None:
self.transforms = Compose([Resize((224, 224)), ToTensor()])
else:
self.transforms = transforms
def __len__(self):
return len(self.x)
def __getitem__(self, idx):
img = Image.open(self.x[idx])
img = img.convert("RGB")
img = self.transforms(img)
return img, torch.tensor([[self.y[idx]]])
以上这篇PyTorch 解决Dataset和Dataloader遇到的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
PyPDF2是一个用于处理PDF文件的Python库，它提供了许多用于读取和操作PDF文件的功能，对于需要处理PDF文件的Python应用程序，PyPDF2是一个非常实用的工具库，本文将给大家详细介绍一下如何通过Python的PyPDF2库提取pdf中的文字,需要的朋友可以参考下2023-05-05
Linux下升级安装python3.8并配置pip及yum的教程
这篇文章主要介绍了Linux下升级安装python3.8并配置pip及yum的教程，本文给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下2020-01-01
in在Python中是操作符，具体来说是成员操作符。这篇文章主要介绍了python 中in 的 用法,需要的朋友可以参考下2019-12-12
今天小编就为大家分享一篇pandas 将索引值相加的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-11-11
今天小编就为大家分享一篇用Python解决x的n次方问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-02-02
本代码介绍了python算法学习中的计数排序实例,代码大家参考使用吧2013-12-12
这篇文章主要介绍了python学生信息管理系统的实现代码，代码简单，复制即可使用,需要的朋友可以参考下2019-12-12
这篇文章主要给大家介绍了关于Python安装Matplotlib包的相关资料,Matplotlib是一个Python 2D绘图库,它以多种硬拷贝格式和跨平台的交互式环境生成出版物质量的图形,需要的朋友可以参考下2023-12-12
详解Python中使用base64模块来处理base64编码的方法
8bit的bytecode经常会被用base64编码格式保存,Python中自带base64模块对base64提供支持,这里我们就来详解Python中使用base64模块来处理base64编码的方法,需要的朋友可以参考下2016-07-07
这篇文章主要为大家详细介绍了pyqt5利用pyqtDesigner实现登录界面，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-03-03
最新评论