forked from acjak/Recon3D
-
Notifications
You must be signed in to change notification settings - Fork 0
/
check_threshold.py
70 lines (61 loc) · 1.88 KB
/
check_threshold.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# Alberto Cereser, September 2017
# DTU Fysik, [email protected]
# This script shows the user a few images, so that she can select which
# threshold value to use
import numpy as np
import matplotlib.pyplot as plt
import sys
'''
Inputs:
Data directory
Modality (1 to plot image array, 2 for image sequence)
Size of the frame used to clean the images
Number of projections to consider
'''
class show_img():
def __init__(self, datadir, modality, frame_size, number_proj):
modality = modality.split(',')
mode = int(modality[0])
frame_size = frame_size.split(',')
frame = int(frame_size[0])
number_proj = number_proj.split(',')
n_proj = int(number_proj[0])
if mode != 1 and mode != 2:
print 'The only modes allowed are 1 and 2.'
else:
A = np.load(datadir + 'dataarray_clean.npy')
if mode == 1:
for oo in range(0,A.shape[2],int(A.shape[2]/n_proj)):
fig = plt.subplots(A.shape[0], A.shape[1])
im_num = 0
for i in range(A.shape[0]):
for j in range(A.shape[1]):
print 'Loading pic', oo, i, j
im_num = im_num + 1
plt.subplot(A.shape[0], A.shape[1], im_num)
A[i,j,oo,0:frame,:] = 0
A[i,j,oo,A.shape[3]-frame:A.shape[3],:] = 0
A[i,j,oo,:,0:frame] = 0
A[i,j,oo,:,A.shape[3]-frame:A.shape[3]] = 0
plt.imshow(A[i,j,oo,:,:])
plt.show()
elif mode == 2:
for oo in range(0,A.shape[2],int(A.shape[2]/n_proj)):
for i in range(A.shape[0]):
for j in range(A.shape[1]):
fig = plt.figure()
plt.imshow(A[i,j,oo,:,:])
plt.show()
if __name__ == "__main__":
if len(sys.argv) != 5:
print "Wrong number of input parameters. Data input should be:\n\
Data directory\n\
Modality (1 to plot image array, 2 for image sequence)\n\
Size of the frame used to clean the images\n\
Number of projections\n\
"
else:
mm = show_img(sys.argv[1],
sys.argv[2],
sys.argv[3],
sys.argv[4])