主要介绍了Pytorch 使用 nii数据做输入数据的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

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