From 28e50b5f3ccc9bc6d81328efd0ba68e8d60d9cd6 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Tue, 18 Feb 2020 11:44:29 +0100
Subject: [PATCH] related to #59
add block name to Concatenate layers
---
src/model_modules/inception_model.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/model_modules/inception_model.py b/src/model_modules/inception_model.py
index 11093df5..1daddd84 100644
--- a/src/model_modules/inception_model.py
+++ b/src/model_modules/inception_model.py
@@ -153,6 +153,7 @@ class InceptionModelBase:
self.number_of_blocks += 1
self.part_of_block = 0
tower_build = {}
+ block_name = f"Block_{self.number_of_blocks}"
for part, part_settings in tower_conv_parts.items():
tower_build[part] = self.create_conv_tower(input_x, **part_settings, **kwargs)
if 'max_pooling' in tower_pool_parts.keys():
@@ -165,7 +166,8 @@ class InceptionModelBase:
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)
- 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
--
GitLab