✏️ [Fix] Some litte typo in module
Browse files- yolo/model/module.py +3 -3
yolo/model/module.py
CHANGED
@@ -189,11 +189,11 @@ class RepNCSPELAN(nn.Module):
|
|
189 |
|
190 |
self.conv1 = Conv(in_channels, part_channels, 1, **kwargs)
|
191 |
self.conv2 = nn.Sequential(
|
192 |
-
RepNCSP(part_channels // 2, process_channels,
|
193 |
Conv(process_channels, process_channels, 3, padding=1, **kwargs),
|
194 |
)
|
195 |
self.conv3 = nn.Sequential(
|
196 |
-
RepNCSP(process_channels, process_channels,
|
197 |
Conv(process_channels, process_channels, 3, padding=1, **kwargs),
|
198 |
)
|
199 |
self.conv4 = Conv(part_channels + 2 * process_channels, out_channels, 1, **kwargs)
|
@@ -245,7 +245,7 @@ class CBLinear(nn.Module):
|
|
245 |
class SPPELAN(nn.Module):
|
246 |
"""SPPELAN module comprising multiple pooling and convolution layers."""
|
247 |
|
248 |
-
def __init__(self, in_channels, out_channels, neck_channels
|
249 |
super(SPPELAN, self).__init__()
|
250 |
neck_channels = neck_channels or out_channels // 2
|
251 |
|
|
|
189 |
|
190 |
self.conv1 = Conv(in_channels, part_channels, 1, **kwargs)
|
191 |
self.conv2 = nn.Sequential(
|
192 |
+
RepNCSP(part_channels // 2, process_channels, neck_args=csp_neck_args, **csp_args),
|
193 |
Conv(process_channels, process_channels, 3, padding=1, **kwargs),
|
194 |
)
|
195 |
self.conv3 = nn.Sequential(
|
196 |
+
RepNCSP(process_channels, process_channels, neck_args=csp_neck_args, **csp_args),
|
197 |
Conv(process_channels, process_channels, 3, padding=1, **kwargs),
|
198 |
)
|
199 |
self.conv4 = Conv(part_channels + 2 * process_channels, out_channels, 1, **kwargs)
|
|
|
245 |
class SPPELAN(nn.Module):
|
246 |
"""SPPELAN module comprising multiple pooling and convolution layers."""
|
247 |
|
248 |
+
def __init__(self, in_channels: int, out_channels: int, neck_channels: Optional[int] = None):
|
249 |
super(SPPELAN, self).__init__()
|
250 |
neck_channels = neck_channels or out_channels // 2
|
251 |
|