
Pytorch 提供了 MyDatasets 功能,用于多通道同时输入不同数据模式。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
目前我正在进行一项项目,采用双通道神经网络架构,每个通道分别接收不同数据进行训练,并且共享相同的标签。最初,我并未设想如何操作,网络上的许多示例都采用单通道结构。即便找到双通道的案例,两个通道的输入数据也往往是相同的。经过反复思考,最终我找到了一个可行的解决方案:通过对torch.utils.data.Datasets模块中的类Dataset类进行重新定义来实现多输入的效果,例如修改class MyDataset(data.Dataset): def __init__(self, data1,data2, labels): self.data1= data1 self.dat
全部评论 (0)
还没有任何评论哟~


