Deactivated pooling and upsampling layers
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -65,13 +65,13 @@ class Encoder(keras.Model):
|
|||||||
result = self.conv1(inputs)
|
result = self.conv1(inputs)
|
||||||
result = self.conv1_a(result)
|
result = self.conv1_a(result)
|
||||||
# result = self.dropout(result)
|
# result = self.dropout(result)
|
||||||
result = self.pool1(result)
|
# result = self.pool1(result)
|
||||||
result = self.conv3(result)
|
result = self.conv3(result)
|
||||||
result = self.conv3_a(result)
|
result = self.conv3_a(result)
|
||||||
result = self.pool3(result)
|
# result = self.pool3(result)
|
||||||
result = self.conv4(result)
|
result = self.conv4(result)
|
||||||
result = self.conv4_a(result)
|
result = self.conv4_a(result)
|
||||||
result = self.pool4(result)
|
# result = self.pool4(result)
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
@ -107,13 +107,13 @@ class Decoder(keras.Model):
|
|||||||
"""See base class."""
|
"""See base class."""
|
||||||
result = self.deconv1(inputs)
|
result = self.deconv1(inputs)
|
||||||
result = self.deconv1_a(result)
|
result = self.deconv1_a(result)
|
||||||
result = self.upsample1(result)
|
# result = self.upsample1(result)
|
||||||
result = self.deconv2(result)
|
result = self.deconv2(result)
|
||||||
result = self.deconv2_a(result)
|
result = self.deconv2_a(result)
|
||||||
result = self.upsample2(result)
|
# result = self.upsample2(result)
|
||||||
result = self.deconv3(result)
|
result = self.deconv3(result)
|
||||||
result = self.deconv3_a(result)
|
result = self.deconv3_a(result)
|
||||||
result = self.upsample3(result)
|
# result = self.upsample3(result)
|
||||||
result = self.deconv5(result)
|
result = self.deconv5(result)
|
||||||
result = k.sigmoid(result)
|
result = k.sigmoid(result)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user