Как кормить изображения nifti в нейронной сети Conv (Unet)

Question

Для моей диссертации у меня есть 16 изображений nifti MRI (Dim: 162×162×192). Мой план - работать с нейронной сетью Conv, и я разделил 10 и 6 изображений как обучающие и тестовые изображения. Проблема в том, как подавать изображения nifti в Unet в качестве изображения поезда для ввода и вывода, а также как подавать тестовые изображения? Спасибо.

Answer 1

Вы можете использовать TorchIO. Вот блокнот Google Colab, в котором изображения NIfTI загружаются в UNet для трехмерной сегментации мозга.