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

__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