Spaces:
Sleeping
Sleeping
✏️ [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 |
|