如何用Pytorch将nii格式数据作为输入进行操作？

本文共计861个文字，预计阅读时间需要4分钟。

使用Pix2Pix-GAN进行医学图像合成的过程中，若将NIfTI数据直接转换为PNG格式，会损失大量信息。因为PNG格式图像的灰度值只有256级，而NIfTI医学图像的灰度值范围通常更广。直接使用NIfTI数据作为输入，效果会更好。

使用pix2pix-gan做医学图像合成的时候，如果把nii数据转成png格式会损失很多信息，以为png格式图像的灰度值有256阶，因此直接使用nii的医学图像做输入会更好一点。

但是Pythorch中的Dataloader是不能直接读取nii图像的，因此加一个CreateNiiDataset的类。

先来了解一下pytorch中读取数据的主要途径——Dataset类。在自己构建数据层时都要基于这个类，类似于C++中的虚基类。