Pytorch 定义MyDatasets实现多通道分别输入不同数据方式
最近在做一个项目,用双通道神经网络,每个通道输入不同数据训练,具有相同label。开始没想到如何实现,网上很多例子都是单通道,即便找到双通道的例子,两个通道的输入也相同。
最后,终于想到了一个办法。多输入和单输入其实是一样的,只需要重新改写torch.utils.data.Datasets.需要改写classDataset里面的init、len和getitem
一个例子:
classMyDataset(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] returnimg1,img2,target def__len__(self): returnlen(self.data1)#我的例子中len(self.data1)=len(self.data2)
以上这篇Pytorch定义MyDatasets实现多通道分别输入不同数据方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。