Neuroimaging-Example/src/const_process_data.py

import os

input_shape: tuple = (4, 80, 96, 64)
# The paper is using (4, 160, 192, 128) with a batch size of 1 to fit the GPU memory.
# We will use a smaller image with a larger batch size because of time & memory issues.

output_channels: int = 3
base_path = os.getcwd() + "/data/raw/*/*"
file_type = '.nii.gz'