- 博客(1)
- 收藏
- 关注
原创 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
980
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人