Pytorch 定义MyDatasets实现多通道分别输入不同数据方式


Posted in Python onJanuary 15, 2020

最近在做一个项目,用双通道神经网络,每个通道输入不同数据训练,具有相同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实现多通道分别输入不同数据方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
对于Python的Django框架使用的一些实用建议
Apr 03 Python
Python fileinput模块使用实例
May 28 Python
python统计日志ip访问数的方法
Jul 06 Python
Python学习之Anaconda的使用与配置方法
Jan 04 Python
pandas series序列转化为星期几的实例
Apr 11 Python
如何优雅地改进Django中的模板碎片缓存详解
Jul 04 Python
python匿名函数用法实例分析
Aug 03 Python
django基于存储在前端的token用户认证解析
Aug 06 Python
让Django的BooleanField支持字符串形式的输入方式
May 20 Python
解决pytorch下出现multi-target not supported at的一种可能原因
Feb 06 Python
python 中的@运算符使用
May 26 Python
Python中Schedule模块使用详解 周期任务神器
Apr 19 Python
pytorch构建多模型实例
Jan 15 #Python
利用Pytorch实现简单的线性回归算法
Jan 15 #Python
pytorch实现线性拟合方式
Jan 15 #Python
Python 支持向量机分类器的实现
Jan 15 #Python
pytorch-神经网络拟合曲线实例
Jan 15 #Python
Pytorch中的VGG实现修改最后一层FC
Jan 15 #Python
详解Python3 中的字符串格式化语法
Jan 15 #Python
You might like
php预定义变量使用帮助(带实例)
2013/10/30 PHP
php结合js实现点击超链接执行删除确认操作
2014/10/31 PHP
php检测apache mod_rewrite模块是否安装的方法
2015/03/14 PHP
js loading加载效果实现代码
2009/11/24 Javascript
jquery的冒泡事件的阻止与允许(三种实现方法)
2013/02/01 Javascript
Javascript跨域请求的4种解决方式
2013/03/17 Javascript
原始XMLHttpRequest方法详情回顾
2013/11/28 Javascript
js监听滚动条滚动事件使得某个标签内容始终位于同一位置
2014/01/24 Javascript
js获取ajax返回值代码
2014/04/30 Javascript
javascript单例模式的简单实现方法
2015/07/25 Javascript
JavaScript实现跨浏览器的添加及删除事件绑定函数实例
2015/08/04 Javascript
JavaScript的this关键字的理解
2016/06/18 Javascript
AngularJS实现tab选项卡的方法详解
2017/07/05 Javascript
vue单页缓存存在的问题及解决方案(小结)
2018/09/25 Javascript
优雅的在React项目中使用Redux的方法
2018/11/10 Javascript
ES6 对象的新功能与解构赋值介绍
2019/02/05 Javascript
微信小程序学习笔记之本地数据缓存功能详解
2019/03/29 Javascript
JS中自定义事件的使用与触发操作实例分析
2019/11/01 Javascript
Python用GET方法上传文件
2015/03/10 Python
python计算圆周率pi的方法
2015/07/11 Python
在Python的Django框架中创建和使用模版
2015/07/15 Python
Python贪吃蛇游戏编写代码
2020/10/26 Python
python中reduce()函数的使用方法示例
2017/09/29 Python
Windows下安装Django框架的方法简明教程
2018/03/28 Python
python3+PyQt5实现自定义流体混合窗口部件
2018/04/24 Python
Redis使用watch完成秒杀抢购功能的代码
2018/05/07 Python
python3实现zabbix告警推送钉钉的示例
2019/02/20 Python
Python socket非阻塞模块应用示例
2019/09/12 Python
python 爬取马蜂窝景点翻页文字评论的实现
2020/01/20 Python
使用数据结构给女朋友写个Html5走迷宫游戏
2019/11/26 HTML / CSS
德国自行车商店:Tretwerk
2019/06/21 全球购物
毕业自我鉴定总结
2014/03/24 职场文书
教师师德演讲稿
2014/05/06 职场文书
2015年七一建党节慰问信
2015/03/23 职场文书
南极大冒险观后感
2015/06/05 职场文书
大学开学感言
2015/08/01 职场文书