From 721ad2aa36e48f0e5b58f26448f7c1762797af91 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 21 Aug 2020 17:22:24 +0200
Subject: [PATCH] Adjusted UNet to changes in input layers.py
---
pysegcnn/core/models.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/pysegcnn/core/models.py b/pysegcnn/core/models.py
index df3c3a6..c04fa4c 100644
--- a/pysegcnn/core/models.py
+++ b/pysegcnn/core/models.py
@@ -26,8 +26,8 @@ import torch.nn as nn
import torch.optim as optim
# locals
-from pysegcnn.core.layers import (Encoder, Decoder, Conv2dPool, Conv2dUnpool,
- Conv2dSame)
+from pysegcnn.core.layers import (Encoder, Decoder, ConvBnReluMaxPool,
+ ConvBnReluMaxUnpool, Conv2dSame)
# module level logger
LOGGER = logging.getLogger(__name__)
@@ -269,11 +269,11 @@ class UNet(Network):
self.epoch = 0
# construct the encoder
- self.encoder = Encoder(filters=self.filters, block=Conv2dPool,
+ self.encoder = Encoder(filters=self.filters, block=ConvBnReluMaxPool,
**kwargs)
# construct the decoder
- self.decoder = Decoder(filters=self.filters, block=Conv2dUnpool,
+ self.decoder = Decoder(filters=self.filters, block=ConvBnReluMaxUnpool,
skip=skip, **kwargs)
# construct the classifier
--
GitLab