Я пытаюсь загрузить свои данные, используя класс ImageDataGenerator в keras, но у меня возникают проблемы, поскольку файлы изображений не являются стандартными файлами изображений jpeg / png, а скорее файлами nii.gz.Я нашел этот репозиторий github 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)