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基于Tkinter的HelloWorld入门实例
Jun 17 Python
python实现八大排序算法(1)
Sep 14 Python
python对日志进行处理的实例代码
Oct 06 Python
Python常用的json标准库
Feb 19 Python
使用python3调用wxpy模块监控linux日志并定时发送消息给群组或好友
Jun 05 Python
如何在Django配置文件里配置session链接
Aug 06 Python
Python爬虫 urllib2的使用方法详解
Sep 23 Python
使用Python制作一个打字训练小工具
Oct 01 Python
基于Django signals 信号作用及用法详解
Mar 28 Python
python实现飞船游戏的纵向移动
Apr 24 Python
python Xpath语法的使用
Nov 26 Python
python疲劳驾驶困倦低头检测功能的实现
Apr 04 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
vBulletin HACK----关于排版的两个HACK
2006/10/09 PHP
Ha0k 0.3 PHP 网页木马修改版
2009/10/11 PHP
PHP Parse Error: syntax error, unexpected $end 错误的解决办法
2012/06/05 PHP
基于swoole实现多人聊天室
2018/06/14 PHP
JS 遮照层实现代码
2010/03/31 Javascript
一个JQuery操作Table的代码分享
2012/03/30 Javascript
js获取当月最后一天实例代码
2013/11/19 Javascript
jquery中val()方法是从最后一个选项往前读取的
2015/09/06 Javascript
js实现兼容IE、Firefox的图片缩放代码
2015/12/08 Javascript
jquery中用函数来设置css样式
2016/12/22 Javascript
jQuery日程管理控件glDatePicker用法详解
2017/03/29 jQuery
jquery dataTable 后台加载数据并分页实例代码
2017/06/07 jQuery
fullpage.js最后一屏滚动方式
2018/02/06 Javascript
深入理解nodejs搭建静态服务器(实现命令行)
2019/02/05 NodeJs
js实现for循环跳过undefined值示例
2019/07/02 Javascript
Vue中img的src是动态渲染时不显示的解决
2019/11/14 Javascript
vue iview实现动态新增和删除
2020/06/17 Javascript
编写Python脚本抓取网络小说来制作自己的阅读器
2015/08/20 Python
Python3 批量扫描端口的例子
2019/07/25 Python
Flask框架单例模式实现方法详解
2019/07/31 Python
彻底搞懂 python 中文乱码问题(深入分析)
2020/02/28 Python
Python基于pandas爬取网页表格数据
2020/05/11 Python
Python利用Pillow(PIL)库实现验证码图片的全过程
2020/10/04 Python
Sephora丝芙兰菲律宾官方网站:购买化妆品和护肤品
2017/04/05 全球购物
匡威比利时官网:Converse Belgium
2017/04/13 全球购物
Ted Baker美国官网:英国时尚品牌
2018/10/29 全球购物
关于是否需要写商业计划书
2014/02/07 职场文书
学习三严三实对照检查材料思想汇报
2014/09/22 职场文书
退休党员个人对照检查材料思想汇报
2014/09/29 职场文书
2014年超市员工工作总结
2014/11/18 职场文书
关于践行三严三实的心得体会
2016/01/05 职场文书
Python 中数组和数字相乘时的注意事项说明
2021/05/10 Python
详细介绍python类及类的用法
2021/05/31 Python
Python使用海龟绘图实现贪吃蛇游戏
2021/06/18 Python
python读取mnist数据集方法案例详解
2021/09/04 Python
Python尝试实现蒙特卡罗模拟期权定价
2022/04/21 Python