Skip to content
Snippets Groups Projects
Commit 28e50b5f authored by Felix Kleinert's avatar Felix Kleinert
Browse files

related to #59

add block name to Concatenate layers
parent 029ffbb0
Branches
Tags
2 merge requests!59Develop,!48Felix issue059 refac naming of concat layers in inception
Pipeline #29819 passed
...@@ -153,6 +153,7 @@ class InceptionModelBase: ...@@ -153,6 +153,7 @@ class InceptionModelBase:
self.number_of_blocks += 1 self.number_of_blocks += 1
self.part_of_block = 0 self.part_of_block = 0
tower_build = {} tower_build = {}
block_name = f"Block_{self.number_of_blocks}"
for part, part_settings in tower_conv_parts.items(): for part, part_settings in tower_conv_parts.items():
tower_build[part] = self.create_conv_tower(input_x, **part_settings, **kwargs) tower_build[part] = self.create_conv_tower(input_x, **part_settings, **kwargs)
if 'max_pooling' in tower_pool_parts.keys(): if 'max_pooling' in tower_pool_parts.keys():
...@@ -165,7 +166,8 @@ class InceptionModelBase: ...@@ -165,7 +166,8 @@ class InceptionModelBase:
tower_build['maxpool'] = self.create_pool_tower(input_x, **tower_pool_parts, **kwargs) tower_build['maxpool'] = self.create_pool_tower(input_x, **tower_pool_parts, **kwargs)
tower_build['avgpool'] = self.create_pool_tower(input_x, **tower_pool_parts, **kwargs, max_pooling=False) tower_build['avgpool'] = self.create_pool_tower(input_x, **tower_pool_parts, **kwargs, max_pooling=False)
block = keras.layers.concatenate(list(tower_build.values()), axis=3) block = keras.layers.concatenate(list(tower_build.values()), axis=3,
name=block_name+"_Co")
return block return block
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment