henry000 commited on
Commit
84ce222
·
1 Parent(s): 9536dc3

✏️ [Fix] Some litte typo in module

Browse files
Files changed (1) hide show
  1. 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, csp_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, csp_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,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=Optional[int]):
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