Add add_shared_layer()

jokame · Nov 25, 2015 · 2685134 · 2685134
1 parent 138e77d
commit 2685134
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/keras/layers/core.py b/keras/layers/core.py
@@ -1365,3 +1365,10 @@ def set_previous(self, layer):
         self.previous = layer
 
 
+def add_shared_layer(layer, inputs):
+    input_layers = [l.layers[-1] for l in inputs]
+    s = Siamese(layer, input_layers, merge_mode=None)
+    for i in range(len(inputs)):
+        sh = SiameseHead(i)
+        inputs[i].add(s)
+        inputs[i].add(sh)