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
投稿猜你喜欢
- Sun公司和Zmanda在3月18号声称他们正在一起合作，为MySQL Enterprise用户交付一种全面、全局数据备份与恢复解决方案。现
- 我就废话不多说了，还是直接看代码吧！import matha=1;//边1b=1;//边2c=math.sqrt(2);//边3A=math
- 私有变量表示方法在变量前加上两个下划线的是私有变量。class Teacher(): def __init__(self,nam
- 本文实例讲述了Python pygorithm模块用法。分享给大家供大家参考，具体如下：pygorithm：一个用纯粹python编写的Py
前言千位分隔符，其实就是数字中的逗号。依西方的习惯，人们在数字中加进一个符号，以免因数字位数太多而难以看出它的值。所以人们在数字中，每隔三位- Python中默认安装的ftplib模块定义了FTP类，可用来实现简单的ftp客户端，用于上传或下载文件。ftp登陆连接from ftpli
- getpass模块的使用：在python中实现密码密文需要导入getpass模块，在python中要使用内置模块的话，需要使用import进
- 1、在全局settings文件中配置```MEDIA_URL = '/media/'MEDIA_ROOT = os.path
- 1.requests库简介requests 是 Python 中比较常用的网页请求库，主要用来发送 HTTP 请求，在使用爬虫或测试服务器响
- 那天偶尔看到看到一个小问题：两个不等长列表a=[1,2,3],b=[4,5,6,7]，求它们对应元素的乘积的和。我一开始想到的方法就是选择更
- 说明Django 默认的用户表 auth_user 包含 id, password, last_login, is_superuser, u
- 即使在urlencode之前str.decode(“cp936″).encode(“utf-8″)做了编码转换也是没用的。后来查询手册查到一
- 大家都知道当任务过多，任务量过大时如果想提高效率的一个最简单的方法就是用多线程去处理，比如爬取上万个网页中的特定数据，以及将爬取数据和清洗数
- 写了网址规范化后，尚奇公司的柳先生建议再深入讨论一下301转向/重定向。下面就谈谈我所了解的301转向在搜索引擎优化方面的应用。什么是301
- 你知道（X）HTML中最多余的标签中是什么吗？在我看来就是这个<a>标签，不错，就是每个网站使用最多的超级链接标签<a&g
- 阅读Chapter 1 清单Chapter 2 标题总览:不但所有网页都需要有标题,而且如果标记正确的话,他们能为网页设计和易用性
- 如下所示：# -*- coding=utf-8 -*- import urllib2import socketimport timeurls
- 对我当前工程进行全部测试需要花费不少时间。既然有 26 GB 空闲内存，为何不让其发挥余热呢? tmpfs 可以通过把文件系统保
- 详细介绍Scrapy shell的使用Scrapy shell是Scrapy框架提供的一个非常有用的工具，可以帮助开发者快速地测试和调试Sc
- 前言最近遇到了一个问题：我的server和client不是在一个时区，server时区是EDT，即美国东部时区，client，就是我自己的电
