Pytorch 定义MyDatasets实现多通道分别输入不同数据方式
作者：Devin-LQP 发布时间：2022-02-25 20:42:51
标签：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)
来源：https://blog.csdn.net/lqp888888/article/details/80481456
0
投稿猜你喜欢
本文要点：爬虫的基本流程requests模块的使用保存csv可视化分析展示环境介绍python 3.8pycharm 2021专业版 激活码- python实现四舍五入""" 四舍五入 :param
- Http定义了与 服务器的交互方法，其中除了一般我们用的最多的GET,POST 其实还有PUT和DELETE 根据RFC2616标
- python类型提示(type hint)在刷leetcode或者一些官方源码的时候，经常看到如下字样：class Solution:&nb
- 最近几个不错网站被封，让人感觉很不爽，现在既不方便用，也不方便学习参考。正好想到曾经“截图”的事情，其实我认为互联网产品还有个特点，更新换代
- 本文实例讲述了Python3写入文件常用方法。分享给大家供大家参考。具体如下：''''' Creat
- 1. 时间与时区1.1 时间标准UTC，世界标准时间，是现在的时间标准，以原子时计时。GMT，格林威治时间，是以前的时间标准，规定太阳每天经
- 一、分工和流程在土豆网，以项目开发为核心，谁都可以带项目，担任项目经理。一个典型的土豆网项目中，当进入正式开发阶段，通常参与者包括：1名设计
- 参数数量及其作用该函数共有两个参数，分别是key和scope。def get_collection(key, scope=None) Wra
- 可视化单词统计词频统计中文分词项目架构新建一个文件，输入文件的内容，查询此文件中关键字的出现的次数，关键字出现的位置，将所有的文本按照中文分
- 在Python中，数据存储方式分为同步存储和异步存储。同步写入速度比较慢，而爬虫速度比较快，有可能导致数据保存不完整，一部分数据没有入库。而
- 工厂方法（Factory Method）模式又称为虚拟构造器（Virtual Constructor）模式或者多态工厂（Polymorphi
- 在linux安装mysql是一个困难的事情，yum安装一般是安装的mysql5.1，现在经过自己不懈努力终于能用yum安装mysql5.5了
- 在开发高并发系统时，有三把利器用来保护系统：缓存、降级和限流。那么何为限流呢？顾名思义，限流就是限制流量，就像你宽带包了1个G的流量，用完了
- 1.获取所有数据库名: SELECT Name FROM Master..SysDatabases ORDER BY Name2.获取所有表
- 问题你想定义一个函数或者方法，它的一个或多个参数是可选的并且有一个默认值。解决方案定义一个有可选参数的函数是非常简单的，直接在函数定义中给参
- 说到这个话题，我们有个产品叫群组，为什么人们需要群组？简单说，群组就是个圈子，是有共同爱好和话题的人群聚在一起讨论、分享的地方。这个产品的诞
- K最近邻属于一种分类算法，他的解释最容易，近朱者赤，近墨者黑，我们想看一个人是什么样的，看他的朋友是什么样的就可以了。当然其他还牵着到，看哪
- 需求分析“员工刷脸考勤”系统，采用Python语言开发，可以通过摄像头添加员工面部信息，这里就涉及到两个具体的个问题，一个是应该以什么样的数
- IE5及其以后版本支持在CSS中使用expression，用来把CSS属性和Javascript表达式关联起来，这里的CSS属性可以是元素固
