Hugging Face's logo

chrisaldikaraharja
/
TumorSegmentationU-Net

Model card Files
xet
Community
chrisaldikaraharja commited on
Commit
97ae15c
·
verified ·
1 Parent(s): 4ef88cb

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +131 -0
config.json ADDED
@@ -0,0 +1,131 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "U-Net",
3
+ "input_shape": [256, 256, 3],
4
+ "num_classes": 2,
5
+ "optimizer": {
6
+ "type": "Adam",
7
+ "learning_rate": 0.0001
8
+ },
9
+ "loss_function": "binary_crossentropy",
10
+ "activation": "sigmoid",
11
+ "metrics": [
12
+ "dice_coef",
13
+ "mean_io_u"
14
+ ],
15
+ "layers": [
16
+ {
17
+ "type": "Conv2D",
18
+ "filters": 64,
19
+ "kernel_size": [3, 3],
20
+ "activation": "relu",
21
+ "padding": "same"
22
+ },
23
+ {
24
+ "type": "Conv2D",
25
+ "filters": 64,
26
+ "kernel_size": [3, 3],
27
+ "activation": "relu",
28
+ "padding": "same"
29
+ },
30
+ {
31
+ "type": "MaxPooling2D",
32
+ "pool_size": [2, 2]
33
+ },
34
+ {
35
+ "type": "Conv2D",
36
+ "filters": 128,
37
+ "kernel_size": [3, 3],
38
+ "activation": "relu",
39
+ "padding": "same"
40
+ },
41
+ {
42
+ "type": "Conv2D",
43
+ "filters": 128,
44
+ "kernel_size": [3, 3],
45
+ "activation": "relu",
46
+ "padding": "same"
47
+ },
48
+ {
49
+ "type": "MaxPooling2D",
50
+ "pool_size": [2, 2]
51
+ },
52
+ {
53
+ "type": "Conv2D",
54
+ "filters": 256,
55
+ "kernel_size": [3, 3],
56
+ "activation": "relu",
57
+ "padding": "same"
58
+ },
59
+ {
60
+ "type": "Conv2D",
61
+ "filters": 256,
62
+ "kernel_size": [3, 3],
63
+ "activation": "relu",
64
+ "padding": "same"
65
+ },
66
+ {
67
+ "type": "MaxPooling2D",
68
+ "pool_size": [2, 2]
69
+ },
70
+ {
71
+ "type": "Conv2D",
72
+ "filters": 512,
73
+ "kernel_size": [3, 3],
74
+ "activation": "relu",
75
+ "padding": "same"
76
+ },
77
+ {
78
+ "type": "Conv2D",
79
+ "filters": 512,
80
+ "kernel_size": [3, 3],
81
+ "activation": "relu",
82
+ "padding": "same"
83
+ },
84
+ {
85
+ "type": "MaxPooling2D",
86
+ "pool_size": [2, 2]
87
+ },
88
+ {
89
+ "type": "Conv2D",
90
+ "filters": 1024,
91
+ "kernel_size": [3, 3],
92
+ "activation": "relu",
93
+ "padding": "same"
94
+ },
95
+ {
96
+ "type": "Conv2DTranspose",
97
+ "filters": 512,
98
+ "kernel_size": [3, 3],
99
+ "strides": [2, 2],
100
+ "padding": "same"
101
+ },
102
+ {
103
+ "type": "Conv2DTranspose",
104
+ "filters": 256,
105
+ "kernel_size": [3, 3],
106
+ "strides": [2, 2],
107
+ "padding": "same"
108
+ },
109
+ {
110
+ "type": "Conv2DTranspose",
111
+ "filters": 128,
112
+ "kernel_size": [3, 3],
113
+ "strides": [2, 2],
114
+ "padding": "same"
115
+ },
116
+ {
117
+ "type": "Conv2DTranspose",
118
+ "filters": 64,
119
+ "kernel_size": [3, 3],
120
+ "strides": [2, 2],
121
+ "padding": "same"
122
+ },
123
+ {
124
+ "type": "Conv2D",
125
+ "filters": 2,
126
+ "kernel_size": [1, 1],
127
+ "activation": "sigmoid",
128
+ "padding": "same"
129
+ }
130
+ ]
131
+ }