2020
09-29
09-29
Pytorch 使用 nii数据做输入数据的操作
使用pix2pix-gan做医学图像合成的时候,如果把nii数据转成png格式会损失很多信息,以为png格式图像的灰度值有256阶,因此直接使用nii的医学图像做输入会更好一点。但是Pythorch中的Dataloader是不能直接读取nii图像的,因此加一个CreateNiiDataset的类。先来了解一下pytorch中读取数据的主要途径——Dataset类。在自己构建数据层时都要基于这个类,类似于C++中的虚基类。自己构建的数据层包含三个部分classDataset(object):"""An...
继续阅读 >