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