--- a
+++ b/code/augmentation.py
@@ -0,0 +1,37 @@
+"""
+	A utility script for data augmentation 
+	
+	Usage: assign your path string to global variable 'path'
+"""
+
+import SimpleITK as sitk
+import numpy as np
+import csv
+from glob import glob
+import pandas as pd
+import os
+import matplotlib
+import matplotlib.pyplot as plt
+import traceback
+import random
+from PIL import Image
+
+path = '/home/ubuntu/data/train-3d/'
+
+file_list=glob(path+"*true*")
+for file in file_list:
+	e = np.load(file)
+	file_name = file.split('/')[-1]
+	# several flip and rotation cases
+	np.save(path+'1_'+file_name,np.transpose(e,(0,2,1)))
+	np.save(path+'2_'+file_name,np.flipud(e))
+	np.save(path+'3_'+file_name,np.flipud(np.transpose(e,(0,2,1))))
+	np.save(path+'4_'+file_name,np.fliplr(e))
+	np.save(path+'5_'+file_name,np.transpose(np.fliplr(e),(0,2,1)))
+	np.save(path+'6_'+file_name,np.flipud(np.fliplr(e)))
+	np.save(path+'7_'+file_name,np.flipud(np.transpose(np.fliplr(e),(0,2,1))))
+	np.save(path+'8_'+file_name,np.transpose(np.rot90(e,1,(1,2)),(0,2,1)))
+	np.save(path+'9_'+file_name,np.rot90(e,1,(1,2)))
+	np.save(path+'10_'+file_name,np.flipud(np.transpose(np.rot90(e,1,(1,2)),(0,2,1))))
+	np.save(path+'11_'+file_name,np.flipud(np.rot90(e,1,(1,2))))
+