Pytorch 定义MyDatasets实现多通道分别输入不同数据方式
更新时间：2020年01月15日 09:59:40 作者：Devin-LQP
今天小编就为大家分享一篇Pytorch 定义MyDatasets实现多通道分别输入不同数据方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
最近在做一个项目，用双通道神经网络，每个通道输入不同数据训练，具有相同label。开始没想到如何实现，网上很多例子都是单通道，即便找到双通道的例子，两个通道的输入也相同。
最后，终于想到了一个办法。多输入和单输入其实是一样的，只需要重新改写torch.utils.data.Datasets. 需要改写class Dataset里面的init、len和getitem
一个例子：
class MyDataset(data.Dataset):
def __init__(self, data1,data2, labels):
self.data1= data1
self.data2= data2
self.labels = labels # 我的例子中label是一样的，如果你的不同，再增加一个即可
def __getitem__(self, index):
img1,img2, target = self.data1[index], self.data2[index], self.labels[index]
return img1,img2, target
def __len__(self):
return len(self.data1) # 我的例子中len(self.data1) = len(self.data2)
以上这篇Pytorch 定义MyDatasets实现多通道分别输入不同数据方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了在Django的视图中使用form对象的方法,Django是Python丰富多彩的开发框架中最具有人气的一个,需要的朋友可以参考下2015-07-07
一步真实解决AttributeError:‘Upsample‘ object has no attribute‘
这篇文章主要介绍了解决解决AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘的问题,本文给大家介绍的非常想详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-06-06
这篇文章主要介绍了python实现的简单文本类游戏,以两个实例形式分析了python操作文本与字符串的相关技巧,具有一定参考借鉴价值,需要的朋友可以参考下2015-04-04
这篇文章主要介绍了anaconda如何创建和删除环境,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-04-04
这篇文章主要介绍了Python实现的桶排序算法,简单说明了桶排序的概念、原理及优缺点,并结合实例形式演示了Python实现桶排序的方法,需要的朋友可以参考下2017-11-11
一条命令解决mac版本python IDLE不能输入中文问题
本文通过一条命令帮助大家解决mac版本python IDLE无法输入中文问题,需要的朋友可以参考下2018-05-05
这篇文章主要介绍了python实现保存网页到本地示例,需要的朋友可以参考下2014-03-03
这篇文章主要介绍了python文件特定行插入和替换实例详解的相关资料,需要的朋友可以参考下2017-07-07
这篇文章主要介绍了Python Pandas 如何shuffle（打乱）数据，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-07-07
在本篇内容里小编给大家分享的是一篇关于通过自学python能找到工作吗的分析内容，对此有兴趣的朋友们可以参考下。2020-06-06
最新评论