[Python] Python 旋转、flip Nifti三维数据 (Python Rotate and Flip Nifti volume)

最新推荐文章于 2022-11-07 13:24:13 发布

Life_XY

最新推荐文章于 2022-11-07 13:24:13 发布

阅读量1w

点赞数 6

分类专栏： Python 图像处理算法

本文链接：https://blog.csdn.net/yangyangyang20092010/article/details/54606152

版权

本文介绍了使用NiBabel库在Python中旋转和翻转Nifti三维数据的方法。需要注意的是，这种变换仅改变头部的仿射矩阵信息，实际数据不受影响。在ITK-Snap加载.nii文件时，会根据仿射矩阵进行转换。然而，这种变换在Matlab的Nifti工具箱中无法识别，会导致错误提示。为解决此问题，需要按照仿射矩阵真正地重新切片体积数据。

摘要由CSDN通过智能技术生成

__author__ = 'xin yang'


import os
import nibabel as nib
import numpy as np
import math


src_us_folder = 'G:/Temp/Data/src/us'
src_seg_folder = 'G:/Temp/Data/src/seg'

aug_us_folder = 'G:/Temp/Data/aug/us'
aug_seg_folder = 'G:/Temp/Data/aug/seg'

img_n= 10
rotate_theta = np.array([0, math.pi/2])

# augmentation
aug_cnt = 0
for k in range(img_n):
    src_us_file = os.path.join(src_us_folder, (str(k) + '.nii'))
    src_seg_file = os.path.join(src_seg_folder, (str(k) + '_seg.nii'))
    # load .nii files
    src_us_vol = nib.load(src_us_file)
    src_seg_vol = nib.load(src_seg_file)
    # volume data
    us_vol_data = src_us_vol.get_data()
    us_vol_data = (np.array(us_vol_data)).astype('uint8')
    seg_vol_data = src_seg_vol.get_data()
    seg_vol_data = (np.array(seg_vol_data)).astype('uint8')
    # get refer