Python 旋转翻转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_