-
Notifications
You must be signed in to change notification settings - Fork 0
/
Evaluation.py
62 lines (46 loc) · 1.66 KB
/
Evaluation.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
import subprocess
import Config
import numpy as np
import glob
import tifffile
import Convert
import sys
def evaluation_likelihood(config):
print ("Evaluating Likelihood")
images = tifffile.imread(config.deployLabels)
target = config.getResultFile("likelihood_target.tif")
tifffile.imsave(target, images[config.deployRange])
arr = np.load(config.likelihood)
arr[arr > 0.5] = 255
arr[arr <= 0.5] = 0
arr = arr.astype(np.uint8)
likelihood = config.getResultFile("likelihood.tif")
tifffile.imsave(likelihood, arr[:, np.newaxis, :, :, 1])
p = subprocess.Popen(["java", "-jar", "Evaluation.jar", target, likelihood],
stdout=config.logStream,
stderr=config.logStream)
p.wait()
config.logStream.flush()
def evaluation_final(config):
print ("Evaluating Final Result")
images = tifffile.imread(config.deployLabels)
target = config.getResultFile("final_target.tif")
tifffile.imsave(target, images[config.segmentRange])
arr = Convert.loadMHA(glob.glob(config.getResultFile("final_*.mha")))
arr[arr != 0.0] = 255
arr = arr.astype(np.uint8)
final = config.getResultFile("final.tif")
tifffile.imsave(final, arr)
p = subprocess.Popen(["java", "-jar", "Evaluation.jar", target, final],
stdout=config.logStream,
stderr=config.logStream)
p.wait()
config.logStream.flush()
def evaluation(config):
if "-nl" not in sys.argv:
evaluation_likelihood(config)
if "-nf" not in sys.argv:
evaluation_final(config)
if __name__ == "__main__":
config = Config.load()
evaluation(config)