-
Notifications
You must be signed in to change notification settings - Fork 54
/
model-dynamic.py
119 lines (88 loc) · 4.52 KB
/
model-dynamic.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
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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
from __future__ import absolute_import
from __future__ import print_function
import os
import keras.models as models
from keras.layers.core import Layer, Dense, Dropout, Activation, Flatten, Reshape, Permute
from keras.layers.convolutional import Conv2D, MaxPooling2D, UpSampling2D, Cropping2D
from keras.layers.normalization import BatchNormalization
from keras.layers import Conv2D, Conv2DTranspose
from keras import backend as K
import cv2
import numpy as np
import json
K.set_image_dim_ordering('th')
# weight_decay = 0.0001
from keras.regularizers import l2
class Tiramisu():
def __init__(self):
self.create()
def DenseBlock(self, layers, filters):
model = self.model
for i in range(layers):
model.add(BatchNormalization())
model.add(Activation('relu'))
model.add(Conv2D(filters, kernel_size=(3, 3), padding='same', init="he_uniform", W_regularizer = l2(0.0001)))
model.add(Dropout(0.2))
def TransitionDown(self,filters):
model = self.model
model.add(BatchNormalization())
model.add(Activation('relu'))
model.add(Conv2D(filters, kernel_size=(1, 1), padding='same', init="he_uniform", W_regularizer = l2(0.0001)))
model.add(Dropout(0.2))
model.add(MaxPooling2D(pool_size=(2, 2)))
def TransitionUp(self,filters, input_shape,output_shape):
model = self.model
model.add(Conv2DTranspose(filters,kernel_size=(3, 3), strides=(2, 2),data_format='channels_first', output_shape=output_shape,
padding='same', input_shape=input_shape, init="he_uniform", W_regularizer = l2(0.0001)))
def gfactorCounterDown(self,model_self,growth_factor,block_size,previous_conv_size,block_count=5):
for i in range(block_count):
m = block_size * growth_factor + previous_conv_size
model_self.DenseBlock(growth_factor,m)
model_self.TransitionDown(growth_factor,m)
def gfactorCounterUp(self,model_self,growth_factor,block_size,previous_block_size,previous_conv_size,block_count=5):
# previous_conv_size = 288, since:
# self.DenseBlock(4,288) # 4*12 = 48 + 288 = 336
# self.TransitionDown(288)
for i in range(block_count):
m = block_size * growth_factor + previous_block_size * growth_factor + previous_conv_size
model_self.DenseBlock(growth_factor,m)
model_self.TransitionDown(growth_factor,m)
def create(self):
model = self.model = models.Sequential()
# cropping
# model.add(Cropping2D(cropping=((68, 68), (128, 128)), input_shape=(3, 360,480)))
model.add(Conv2D(48, kernel_size=(3, 3), padding='same', input_shape=(3,224,224), init="he_uniform", W_regularizer = l2(0.0001)))
# (5 * 4)* 2 + 5 + 5 + 1 + 1 +1
# growth_m = 4 * 12
# previous_m = 48
self.gfactorCounterDown(self.model,12,4,48,5)
# self.DenseBlock(4,96) # 4*12 = 48 + 48 = 96
# self.TransitionDown(96)
# self.DenseBlock(4,144) # 4*12 = 48 + 96 = 144
# self.TransitionDown(144)
# self.DenseBlock(4,192) # 4*12 = 48 + 144 = 192
# self.TransitionDown(192)
# self.DenseBlock(4,240)# 4*12 = 48 + 192 = 240
# self.TransitionDown(240)
# self.DenseBlock(4,288) # 4*12 = 48 + 288 = 336
# self.TransitionDown(288)
self.DenseBlock(15,336) # 4 * 12 = 48 + 288 = 336
self.gfactorCounterDown(self.model,12,4,4,288,5)
# self.TransitionUp(384, (384, 7, 7), (None, 384, 14, 14)) # m = 288 + 4x12 + 4x12 = 384.
# self.DenseBlock(4,384)
# self.TransitionUp(336, (336, 14, 14), (None, 336, 28, 28)) #m = 240 + 4x12 + 4x12 = 336
# self.DenseBlock(4,336)
# self.TransitionUp(288, (288, 28, 28), (None, 288, 56, 56)) # m = 192 + 4x12 + 4x12 = 288
# self.DenseBlock(4,288)
# self.TransitionUp(240, (240, 56, 56), (None, 240, 112, 112)) # m = 144 + 4x12 + 4x12 = 240
# self.DenseBlock(4,240)
# self.TransitionUp(192, (192, 112, 112), (None, 192, 224, 224)) # m = 96 + 4x12 + 4x12 = 192
# self.DenseBlock(4,192)
model.add(Conv2D(12, kernel_size=(3, 3), padding='same', init="he_uniform", W_regularizer = l2(0.0001)))
model.add(Reshape((12, 224 * 224)))
model.add(Permute((2, 1)))
model.add(Activation('softmax'))
model.summary()
with open('tiramisu_fc_dense56_model.json', 'w') as outfile:
outfile.write(json.dumps(json.loads(model.to_json()), indent=3))
Tiramisu()