-
Notifications
You must be signed in to change notification settings - Fork 0
/
rescaling.py
53 lines (37 loc) · 1.22 KB
/
rescaling.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# -*- coding: utf-8 -*-
import SimpleITK as sitk
import os
import numpy as np
from datetime import datetime
import argparse
class PreProcess(object):
data_path = None
def __init__(self, data_path):
if data_path != None:
self.data_path = data_path
def rescale(self):
for root, dirs, files in os.walk(self.data_path):
for name in files:
file_path=os.path.join(root, name)
print("file:",file_path)
img = sitk.ReadImage(file_path)
img_arr = sitk.GetArrayFromImage(img)
#max=np.amax(img_arr)
#min=np.amin(img_arr)
#max=2058
#min=0
max=1989
min=-1000
diff=(max-min)/2.0
middle=(max+min)/2.0
img_arr=(img_arr-middle)/diff
print("rescaled")
new_nii = sitk.GetImageFromArray(img_arr)
sitk.WriteImage(new_nii, file_path)
def main(args):
pp = PreProcess(data_path=args.rootdir)
pp.rescale()
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('rootdir', help='Directory of Data')
main(parser.parse_args())