From c104ba6239e81f00443309ca2a2498e41c696b4d Mon Sep 17 00:00:00 2001 From: cszdlt Date: Sun, 29 Dec 2019 00:53:53 +0800 Subject: [PATCH] =?UTF-8?q?=E8=BF=98=E5=8E=9FSAEHD=E6=A8=A1=E5=9E=8B?= =?UTF-8?q?=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- models/Model_SAEHD/Model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index b02f2de..9957561 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -213,7 +213,7 @@ class SAEHDModel(ModelBase): x = Add()([x, x0]) x = LeakyReLU(0.2)(x) - return Conv2D(output_nc, kernel_size=5, padding='same', activation='sigmoid')(x) + return Conv2D(output_nc, kernel_size=1, padding='same', activation='sigmoid')(x) return func @@ -327,7 +327,7 @@ class SAEHDModel(ModelBase): x = Add()([x, x0]) x = LeakyReLU(0.2)(x) - return Conv2D(output_nc, kernel_size=5, padding='same', activation='sigmoid')(x) + return Conv2D(output_nc, kernel_size=1, padding='same', activation='sigmoid')(x) return func