Pytorch 定义MyDatasets实现多通道分别输入不同数据方式
更新时间：2020年01月15日 09:59:40 作者：Devin-LQP
今天小编就为大家分享一篇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)
以上这篇Pytorch 定义MyDatasets实现多通道分别输入不同数据方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家详细介绍了如何基于Python开发一个文件快速搜索工具,可以实现多条件文件搜索并实时搜索状态反馈,需要的可以参考一下2025-03-03
今天小编就为大家分享一篇基于keras 模型、结构、权重保存的实现，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
本文主要介绍了OpenCV仿射变换的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-08-08
这篇文章主要介绍了Pytorch中的backward()多个loss函数用法，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-05-05
os.path模块主要用于文件的属性获取，在编程中经常用到，本文将带你熟悉这个模块并掌握它的用法，感兴趣的朋友跟小编来看看吧2021-10-10
这篇文章主要为大家详细介绍了如何利用Python语言实现Linux系统的CPU以及内存占用，文中的示例代码讲解详细，具有一定的学习价值，需要的可以收藏一下2023-05-05
今天小编就为大家分享一篇使用pyecharts生成Echarts网页的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
这篇文章主要为大家详细介绍了python多进程下实现日志记录按时间分割，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-07-07
python利用re,bs4,requests模块获取股票数据
这篇文章主要介绍了python利用re,bs4,requests模块获取股票数据,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
这篇文章主要介绍了关于Python中的空值问题及解决方案,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-11-11
最新评论