Pytorch 定义定 MyDatasets实现多通道分别输入不同数据方式

Pytorch 定义定 MyDatasets实现多通道分别输入不同数据方式

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

提示：这里可以添加本文要记录的大概内容：

最近在做一个项目，用双通道神经网络，每个通道输入不同数据训练，具有相同label。开始没想到如何实现，网上很多例子 都是单通道，即便找到双通道的例子，两个通道的输入也相同。 最后，终于想到了一个办法。多输入和单输入其实是一样的，只需要重新改写torch.utils.data.Datasets. 需要改写class Dataset里面的init、len和getitem

---

提示：以下是本篇文章正文内容，下面案例可供参考

双通道神经网络

示例：
相关的一个例子：

class MyDataset(data.Dataset): 
def __init__(self