70x70 patch gan

Author: ktaebum

models/deepunet.py

@@ -177,11 +177,6 @@ def forward(self, x):
 
 
 class DeepUNetUpSample(nn.Module):
-    """
-    TODO:
-        attention features will be added
-    """
-
     def __init__(self, in_channels, out_channels, bias=True, dropout=False):
         super(DeepUNetUpSample, self).__init__()
         self.pool = nn.MaxUnpool2d(2, 2)

models/patch_gan.py

@@ -30,14 +30,11 @@ def __init__(self, dim=64, norm='batch', sigmoid=True):
         # 64 x 64
         layers.append(self._building_block(self.dim * 2, self.dim * 4))
 
-        # 32 x 32
-        layers.append(self._building_block(self.dim * 4, self.dim * 8))
-
-        # 31 x 31
+        # 63 x 63
         layers.append(
-            self._building_block(self.dim * 8, self.dim * 8, stride=1))
+            self._building_block(self.dim * 4, self.dim * 8, stride=1))
 
-        # 30 x 30
+        # 62 x62
         layers.append(
             nn.Sequential(
                 nn.Conv2d(self.dim * 8, 1, 4, 1, 1),