lesson1-UNet Model

goodpupil · Dec 5, 2019 · 6bcbe3b · 6bcbe3b
1 parent 2610e68
commit 6bcbe3b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Pytorch-Seg/lesson-1/unet_parts.py b/Pytorch-Seg/lesson-1/unet_parts.py
@@ -12,10 +12,10 @@ class DoubleConv(nn.Module):
     def __init__(self, in_channels, out_channels):
         super().__init__()
         self.double_conv = nn.Sequential(
-            nn.Conv2d(in_channels, out_channels, kernel_size=3, padding=1),
+            nn.Conv2d(in_channels, out_channels, kernel_size=3, padding=0),
             nn.BatchNorm2d(out_channels),
             nn.ReLU(inplace=True),
-            nn.Conv2d(out_channels, out_channels, kernel_size=3, padding=1),
+            nn.Conv2d(out_channels, out_channels, kernel_size=3, padding=0),
             nn.BatchNorm2d(out_channels),
             nn.ReLU(inplace=True)
         )
@@ -73,4 +73,4 @@ def __init__(self, in_channels, out_channels):
         self.conv = nn.Conv2d(in_channels, out_channels, kernel_size=1)
 
     def forward(self, x):
-        return self.conv(x)
+        return self.conv(x)