--- a +++ b/utils/niftimanager.py @@ -0,0 +1,34 @@ + +import nibabel as nib +import matplotlib.pyplot as plt + +def load_nifti(file_path): + ''' + Load the NIfTI image and access the image data as a Numpy array. + + Args: + file_path ('str'): Path to the NIfTI file. + + Returns: + data_array ('np.array'): Numpy array representing the image data. + nii_image: Loaded NIfTI image object. + ''' + nii_image = nib.load(file_path) + data_array = nii_image.get_fdata() + + return data_array, nii_image + +def show_nifti(file_data, title, slice=25): + ''' + Display a single slice from the NIfTI volume. + + Args: + file_data ('np.array'): Numpy array representing the image data. + title ('str'): Title for the plot. + slice ('int'): Slice index to display. + ''' + plt.imshow(file_data[slice, :, :], cmap='gray') + plt.title(title) + # plt.colorbar() + plt.axis('off') + plt.show() \ No newline at end of file