如何使用ADNI数据集将ImageDataGenerator与nii / NIFTI文件一起使用

Question

我正在尝试使用keras的ImageDataGenerator类加载数据，但是由于图像文件不是标准的jpeg / png图像文件而是nii.gz文件，因此我遇到了麻烦。我找到了这个github repo https://github.com/sremedios/nifti_image_generator/blob/master/utils/nifti_image.py，但是输出的尺寸不匹配，并且

train_generator.next()

引发

错误

ValueError: could not broadcast input array from shape (233,189) into shape (197,233,189,1)

Answer 1

为了使用Nii文件，该文件是MR，CT等医学图像的标准格式。 您必须使用nibabel库来读取医学影像数据集。
DLTK库也可以从Google获得，用于医学成像。