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程序员开发中常犯的10个错误
Jul 07 Python
Python读写Json涉及到中文的处理方法
Sep 12 Python
python socket网络编程之粘包问题详解
Apr 28 Python
Pycharm配置远程调试的方法步骤
Dec 17 Python
Python面向对象程序设计中类的定义、实例化、封装及私有变量/方法详解
Feb 28 Python
Python动态导入模块和反射机制详解
Feb 18 Python
python3中datetime库,time库以及pandas中的时间函数区别与详解
Apr 16 Python
python 制作python包,封装成可用模块教程
Jul 13 Python
Python+Selenium随机生成手机验证码并检查页面上是否弹出重复手机号码提示框
Sep 21 Python
python爬虫中的url下载器用法详解
Nov 30 Python
python中_del_还原数据的方法
Dec 09 Python
python爬取新闻门户网站的示例
Apr 25 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
Body是什么,该怎么喝出咖啡里的口感
2021/03/03 咖啡文化
同时提取多条新闻中的文本一例
2006/10/09 PHP
PHP 伪静态技术原理以及突破原理实现介绍
2013/07/12 PHP
php中fsockopen用法实例
2015/01/05 PHP
PHP获取本周所有日期或者最近七天所有日期的方法
2018/06/20 PHP
PHP获取真实IP及IP模拟方法解析
2020/11/24 PHP
jQuery LigerUI 插件介绍及使用之ligerDrag和ligerResizable示例代码打包
2011/04/06 Javascript
使用delegate方法为一个tr标签加一个链接
2014/06/27 Javascript
jquery实现的代替传统checkbox样式插件
2015/06/19 Javascript
关于bootstrap日期转化,bootstrap-editable的简单使用,bootstrap-fileinput的使用详解
2017/05/12 Javascript
关于js中的鼠标事件总结
2017/07/11 Javascript
webpack vue项目开发环境局域网访问方法
2018/03/20 Javascript
js动态设置select下拉菜单的默认选中项实例
2018/08/21 Javascript
Nuxt.js开启SSR渲染的教程详解
2018/11/30 Javascript
Element-UI中Upload上传文件前端缓存处理示例
2019/02/21 Javascript
浅谈JavaScript面向对象--继承
2019/03/20 Javascript
vue前端框架—Mint UI详解(更适用于移动端)
2019/04/30 Javascript
微信小程序用户授权、位置授权及获取微信绑定手机号
2019/07/18 Javascript
搭建Vue从Vue-cli到router路由护卫的实现
2019/11/14 Javascript
利用PHP实现递归删除链表元素的方法示例
2020/10/23 Javascript
python二分法实现实例
2013/11/21 Python
Python操作使用MySQL数据库的实例代码
2017/05/25 Python
Python实现输出某区间范围内全部素数的方法
2018/05/02 Python
解决TensorFlow模型恢复报错的问题
2020/02/06 Python
Python3操作MongoDB增册改查等方法详解
2020/02/10 Python
记录模型训练时loss值的变化情况
2020/06/16 Python
几款主流好用的富文本编辑器(所见即所得常用编辑器)介绍
2021/03/17 Javascript
英国排名第一的最新设计师品牌手表独立零售商:TIC Watches
2016/09/24 全球购物
学习委员自我鉴定
2014/01/13 职场文书
幼儿园教师自我鉴定
2014/03/20 职场文书
2014年扶贫工作总结
2014/11/18 职场文书
摘录式读书笔记
2015/07/01 职场文书
大学校园餐饮创业计划书
2019/08/07 职场文书
解决SpringBoot文件上传临时目录找不到的问题
2021/07/01 Java/Android
避坑之 JavaScript 中的toFixed()和正则表达式
2022/04/19 Javascript
PostgreSQL常用字符串分割函数整理汇总
2022/07/07 PostgreSQL