Pytorch 定义MyDatasets实现多通道分别输入不同数据
最近在做一个项目,用的双通道神经网络,每个通道输入不同数据训练,标签是一样的,开始没想通怎么实现,网上很多例子都是单通道,即便找到双通道的例子,两个通道的输入数据也相同。最后,终于想到了一个办法,其实多输入和单输入是一样的,我们只需要重新改写torch.utils.data.Datasets. 需要改写的是class Dataset里面的__init__、__len__和__getitem__...
原创
2018-05-28 14:05:42 ·
10538 阅读 ·
16 评论