Update model/models.py
Browse files- model/models.py +4 -4
model/models.py
CHANGED
@@ -71,8 +71,8 @@ class Selayer(nn.Module):
|
|
71 |
def __init__(self, inplanes):
|
72 |
super(Selayer, self).__init__()
|
73 |
self.global_avgpool = nn.AdaptiveAvgPool2d(1)
|
74 |
-
self.conv1 = nn.Conv2d(
|
75 |
-
self.conv2 = nn.Conv2d(inplanes // 16,
|
76 |
self.relu = nn.ReLU(inplace=True)
|
77 |
self.sigmoid = nn.Sigmoid()
|
78 |
|
@@ -89,8 +89,8 @@ class SelayerSpectr(nn.Module):
|
|
89 |
def __init__(self, inplanes):
|
90 |
super(SelayerSpectr, self).__init__()
|
91 |
self.global_avgpool = nn.AdaptiveAvgPool2d(1)
|
92 |
-
self.conv1 = SpectralNorm(nn.Conv2d(
|
93 |
-
self.conv2 = SpectralNorm(nn.Conv2d(inplanes // 16,
|
94 |
self.relu = nn.ReLU(inplace=True)
|
95 |
self.sigmoid = nn.Sigmoid()
|
96 |
|
|
|
71 |
def __init__(self, inplanes):
|
72 |
super(Selayer, self).__init__()
|
73 |
self.global_avgpool = nn.AdaptiveAvgPool2d(1)
|
74 |
+
self.conv1 = nn.Conv2d(2, inplanes // 16, kernel_size=1, stride=1)
|
75 |
+
self.conv2 = nn.Conv2d(inplanes // 16, 2, kernel_size=1, stride=1)
|
76 |
self.relu = nn.ReLU(inplace=True)
|
77 |
self.sigmoid = nn.Sigmoid()
|
78 |
|
|
|
89 |
def __init__(self, inplanes):
|
90 |
super(SelayerSpectr, self).__init__()
|
91 |
self.global_avgpool = nn.AdaptiveAvgPool2d(1)
|
92 |
+
self.conv1 = SpectralNorm(nn.Conv2d(2, inplanes // 16, kernel_size=1, stride=1))
|
93 |
+
self.conv2 = SpectralNorm(nn.Conv2d(inplanes // 16, 2, kernel_size=1, stride=1))
|
94 |
self.relu = nn.ReLU(inplace=True)
|
95 |
self.sigmoid = nn.Sigmoid()
|
96 |
|