-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathgraph.dot
104 lines (84 loc) · 2.45 KB
/
graph.dot
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
digraph G {
label = "Lung Instance Segmentation Workflow"
labelloc="t"
labelfontsize=20.0
subgraph cluster_preprocessing {
preprocess_train
preprocess_val
preprocess_test
label="preprocessing steps";
color=pink;
}
subgraph cluster_input {
subgraph cluster_train {
train_imgs
train_masks
label="train"
}
subgraph cluster_val {
val_imgs
val_masks
label="validation"
}
subgraph cluster_test {
test_imgs
test_masks
label="test"
}
label="input data (lung and corresponding mask images)"
}
subgraph cluster_processed_images {
"val_imgs`";
"test_imgs`";
label="";
color=white;
subgraph cluster_augmented {
"augmented_train_imgs`";
"augmented_train_masks"
}
}
train_imgs -> preprocess_train;
train_masks -> preprocess_train
val_imgs -> preprocess_val;
test_imgs -> preprocess_test;
preprocess_train -> "augmented_train_imgs`";
preprocess_train -> "augmented_train_masks"
preprocess_val -> "val_imgs`";
preprocess_test -> "test_imgs`";
train_imgs [shape=box]
val_imgs [shape=box]
test_imgs [shape=box]
train_masks [shape=box]
val_masks [shape=box]
test_masks [shape=box]
"augmented_train_imgs`" [shape=box]
"val_imgs`" [shape=box]
"test_imgs`" [shape=box]
"augmented_train_masks" [shape=box]
generated_masks [shape=box]
metrics [shape=box]
"study_results.txt" [shape=box]
"study_checkpoint.pkl" [shape=box]
"model.h5" [shape=box]
train_model_checkpoint [shape=box]
"augmented_train_imgs`" -> train_model
"val_imgs`" -> train_model
"augmented_train_masks" -> train_model
"val_masks" -> train_model
"augmented_train_imgs`" -> hpo
"val_imgs`" -> hpo
"augmented_train_masks" -> hpo
"val_masks" -> hpo
"test_imgs`" -> predict_masks
predict_masks -> generated_masks
generated_masks -> evaluate
test_masks -> evaluate
"test_imgs`" -> evaluate
evaluate -> metrics
hpo -> "study_results.txt"
hpo -> "study_checkpoint.pkl"
"study_results.txt" -> train_model
train_model -> "model.h5"
train_model -> train_model_checkpoint
"model.h5" -> predict_masks
}