weixin_46548241的博客

原创 pytorch自定义dataset类分别加载多通道图片并合并

pytorch自定义dataset类分别加载多通道图片因为需要处理多通道的传感器信息，所以自定义了一个dataset类来分别同时加载多通道图片，用于同时输入到卷积神经网络中。MyDataset介绍我这里是打算同时四个通道分别加载图片，并且预处理之后将四张图片合并。也就是说输入之前图片尺寸是4*(3,image_size,image_size)（如果加载灰度图的话就是(1,image_size,image_size)），通过MyDataset包装之后返回的张量大小为(34,image_size,ima