From 0b9447b3787e86935a732ff8e3863efc2736dd8a Mon Sep 17 00:00:00 2001
From: Ienco Dino <dino.ienco@irstea.fr>
Date: Wed, 9 Dec 2020 19:42:28 +0100
Subject: [PATCH] Update TwoBranchCNN1D.py
---
TwoBranchCNN1D.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/TwoBranchCNN1D.py b/TwoBranchCNN1D.py
index 97ea57c..3e8a2a9 100644
--- a/TwoBranchCNN1D.py
+++ b/TwoBranchCNN1D.py
@@ -39,15 +39,15 @@ def getBranch(input_layer, n_filters, dropOut, suffix):
-def build_model(self, input_shape1, input_shape2, nb_classes):
+def build_model(input_shape1, input_shape2, nb_classes):
n_feature_maps = 128
dropOut = 0.2
input_layer1 = tfk.layers.Input(name="input1", shape=input_shape1)
input_layer2 = tfk.layers.Input(name="input2", shape=input_shape2)
- features1 = self.getBranch(input_layer1, n_feature_maps, dropOut, "input1")
- features2 = self.getBranch(input_layer2, n_feature_maps, dropOut, "input2")
+ features1 = getBranch(input_layer1, n_feature_maps, dropOut, "input1")
+ features2 = getBranch(input_layer2, n_feature_maps, dropOut, "input2")
features = tfk.Concatenation(name="concat_layer")([features1, features2])
dense_layer = tfk.layers.Dense(512, activation='relu')(features)
--
GitLab