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之使用Python查询更新数据库
Nov 25 Python
python装饰器与递归算法详解
Feb 18 Python
Python 模拟购物车的实例讲解
Sep 11 Python
Python入门之三角函数atan2()函数详解
Nov 08 Python
Python解析命令行读取参数--argparse模块使用方法
Jan 23 Python
用Python写脚本,实现完全备份和增量备份的示例
Apr 29 Python
基于python及pytorch中乘法的使用详解
Dec 27 Python
浅谈tensorflow中张量的提取值和赋值
Jan 19 Python
Python线程协作threading.Condition实现过程解析
Mar 12 Python
Python自动巡检H3C交换机实现过程解析
Aug 14 Python
用python批量移动文件
Jan 14 Python
Python爬虫基础之简单说一下scrapy的框架结构
Jun 26 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 Static关键字实用方法
2010/06/04 PHP
java模拟PHP的pack和unpack类
2016/04/13 PHP
Web 前端设计模式--Dom重构 提高显示性能
2010/10/22 Javascript
jQuery选择器全集详解
2014/11/24 Javascript
jquery 操作css样式、位置、尺寸方法汇总
2014/11/28 Javascript
AngularJS HTML编译器介绍
2014/12/06 Javascript
node.js中的fs.lchownSync方法使用说明
2014/12/16 Javascript
JS实现的论坛Ajax打分效果完整实例
2015/10/31 Javascript
jQuery简单实现title提示效果示例
2016/08/01 Javascript
浅谈js中的引用和复制(传值和传址)
2016/09/18 Javascript
微信小程序实现皮肤功能(夜间模式)
2017/06/18 Javascript
Node 自动化部署的方法
2017/10/17 Javascript
js保留两位小数方法总结
2018/01/31 Javascript
vue2.0 实现导航守卫(路由守卫)
2018/05/21 Javascript
webpack 从指定入口文件中提取公共文件的方法
2018/11/13 Javascript
JQuery中queue方法用法示例
2019/01/31 jQuery
Vue实现导航栏点击当前标签变色功能
2020/08/19 Javascript
Element InputNumber 计数器的实现示例
2020/08/03 Javascript
JS获取一个字符串中指定字符串第n次出现的位置
2021/02/10 Javascript
Python编程使用NLTK进行自然语言处理详解
2017/11/16 Python
在Pandas中处理NaN值的方法
2019/06/25 Python
Python中最好用的命令行参数解析工具(argparse)
2019/08/23 Python
python向图片里添加文字
2019/11/26 Python
paramiko使用tail实时获取服务器的日志输出详解
2020/12/06 Python
python利用pytesseract 实现本地识别图片文字
2020/12/14 Python
Python用access判断文件是否被占用的实例方法
2020/12/17 Python
我想声明一个指针并为它分配一些空间, 但却不行。这些代码有什么 问题?char *p; *p = malloc(10);
2016/10/06 面试题
一份软件工程师的面试试题
2016/02/01 面试题
函授自我鉴定范文
2014/02/06 职场文书
《阳光》教学反思
2014/02/23 职场文书
群教个人对照检查材料
2014/08/20 职场文书
安全例会汇报材料
2014/08/23 职场文书
党支部党的群众路线对照检查材料
2014/09/24 职场文书
2014年学生会工作总结范文
2014/11/07 职场文书
有关花店创业的计划书模板
2019/08/27 职场文书
美国运营商 T-Mobile 以 117.83Mb/s 的速度排第一位
2022/04/21 数码科技