Загрузите 51 массив размером (124, 512, 512) в кадр данных - PullRequest
0 голосов
/ 27 марта 2020

Мне нужно обучить нейронную сеть с изображениями.

Для этого мне нужно использовать 51 файл в форме массивов: (124, 512, 512) и int16 в качестве типа данных. Изображения имеют размер 512 на 512 и имеют 124 фрагмента тома.

Я не знаю, лучше ли импортировать все это в моей среде Python или оставить ее в своей папке и выполнить операции на файлы с путем к файлу. Это мои 51 имя файла в моей переменной «volumelist» в виде списка строк.

volume/0.nrrd
volume/1.nrrd
volume/10.nrrd
volume/11.nrrd
volume/12.nrrd
volume/13.nrrd

Это мой сценарий после прочтения файлов в моем каталоге и помещения их в список.

scandataset =[]
images = range(len(volumelist))
for image in images:
    # import all files as a list of arrays
    imgscan = sitk.ReadImage(volumelist[image])
    scandataset = sitk.GetArrayFromImage(imgscan)
    print(image)

Как я могу сохранить все 51 файл трехмерных массивов в учебном наборе данных?

...