I have defined a simple model using the keras Model functional API. One of its layers is a fully sequential model, so I get a nested layer structure (see images below).
How can I convert this nested layer structure into a flat layer structure? (with a script, not manually...)
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
input_1 (InputLayer) (None, 32, 32, 1) 0
_________________________________________________________________
sequential_1 (Sequential) (None, 8, 8, 12) 720
_________________________________________________________________
flatten_1 (Flatten) (None, 768) 0
_________________________________________________________________
dense_1 (Dense) (None, 1) 769
=================================================================
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
input_1 (InputLayer) (None, 32, 32, 1) 0
_________________________________________________________________
conv2d_1 (Conv2D) (None, 32, 32, 6) 60
_________________________________________________________________
max_pooling2d_1 (MaxPooling2 (None, 16, 16, 6) 0
_________________________________________________________________
conv2d_2 (Conv2D) (None, 16, 16, 6) 330
_________________________________________________________________
max_pooling2d_2 (MaxPooling2 (None, 8, 8, 6) 0
_________________________________________________________________
flatten_1 (Flatten) (None, 384) 0
_________________________________________________________________
dense_1 (Dense) (None, 1) 385
=================================================================
Code to generate nested layer structure:
def create_network_with_one_subnet():
# define subnetwork
subnet = keras.models.Sequential()
subnet.add(keras.layers.Conv2D(6, (3, 3), padding='same'))
subnet.add(keras.layers.MaxPool2D())
subnet.add(keras.layers.Conv2D(12, (3, 3), padding='same'))
subnet.add(keras.layers.MaxPool2D())
#subnet.summary()
# define complete network
input_shape = (32, 32, 1)
net_in = keras.layers.Input(shape=input_shape)
net_out = subnet(net_in)
net_out = keras.layers.Flatten()(net_out)
net_out = keras.layers.Dense(1)(net_out)
net_complete = keras.Model(inputs=net_in, outputs=net_out)
net_complete.compile(loss='binary_crossentropy',
optimizer=keras.optimizers.Adam(lr=0.001),
metrics=['acc'],
)
net_complete.summary()
return net_complete
Ah, it was much easier than expected. Solution from here after googling the right keywords: https://groups.google.com/forum/#!msg/keras-users/lJcVK25YDuc/atB6TfwqBAAJ
def flatten_model(model_nested):
layers_flat = []
for layer in model_nested.layers:
try:
layers_flat.extend(layer.layers)
except AttributeError:
layers_flat.append(layer)
model_flat = keras.models.Sequential(layers_flat)
return model_flat
Slightly better solution for handling nested models with more than one level:
def flatten_model(model_nested):
def get_layers(layers):
layers_flat = []
for layer in layers:
try:
layers_flat.extend(get_layers(layer.layers))
except AttributeError:
layers_flat.append(layer)
return layers_flat
model_flat = tfk.models.Sequential(
get_layers(model_nested.layers)
)
return model_flat
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.