-
Notifications
You must be signed in to change notification settings - Fork 1
/
outputs.py
35 lines (35 loc) · 2.02 KB
/
outputs.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
unet_model_output = [['InputLayer', [(None, 96, 128, 3)], 0],
['Conv2D', (None, 96, 128, 32), 896, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 96, 128, 32), 9248, 'same', 'relu', 'HeNormal'],
['MaxPooling2D', (None, 48, 64, 32), 0, (2, 2)],
['Conv2D', (None, 48, 64, 64), 18496, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 48, 64, 64), 36928, 'same', 'relu', 'HeNormal'],
['MaxPooling2D', (None, 24, 32, 64), 0, (2, 2)],
['Conv2D', (None, 24, 32, 128), 73856, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 24, 32, 128), 147584, 'same', 'relu', 'HeNormal'],
['MaxPooling2D', (None, 12, 16, 128), 0, (2, 2)],
['Conv2D', (None, 12, 16, 256), 295168, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 12, 16, 256), 590080, 'same', 'relu', 'HeNormal'],
['Dropout', (None, 12, 16, 256), 0, 0.3],
['MaxPooling2D', (None, 6, 8, 256), 0, (2, 2)],
['Conv2D', (None, 6, 8, 512), 1180160, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 6, 8, 512), 2359808, 'same', 'relu', 'HeNormal'],
['Dropout', (None, 6, 8, 512), 0, 0.3],
['Conv2DTranspose', (None, 12, 16, 256), 1179904],
['Concatenate', (None, 12, 16, 512), 0],
['Conv2D', (None, 12, 16, 256), 1179904, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 12, 16, 256), 590080, 'same', 'relu', 'HeNormal'],
['Conv2DTranspose', (None, 24, 32, 128), 295040],
['Concatenate', (None, 24, 32, 256), 0],
['Conv2D', (None, 24, 32, 128), 295040, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 24, 32, 128), 147584, 'same', 'relu', 'HeNormal'],
['Conv2DTranspose', (None, 48, 64, 64), 73792],
['Concatenate', (None, 48, 64, 128), 0],
['Conv2D', (None, 48, 64, 64), 73792, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 48, 64, 64), 36928, 'same', 'relu', 'HeNormal'],
['Conv2DTranspose', (None, 96, 128, 32), 18464],
['Concatenate', (None, 96, 128, 64), 0],
['Conv2D', (None, 96, 128, 32), 18464, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 96, 128, 32), 9248, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 96, 128, 32), 9248, 'same', 'relu', 'HeNormal'],
['Conv2D', (None, 96, 128, 23), 759, 'same', 'linear', 'GlorotUniform']]