Fix channel bug in centerpoint Separatehead (#228)

open-mmlab · Dec 13, 2020 · bafee2e · bafee2e
1 parent e5a155e
commit bafee2e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mmdet3d/models/dense_heads/centerpoint_head.py b/mmdet3d/models/dense_heads/centerpoint_head.py
@@ -51,17 +51,19 @@ def __init__(self,
  classes, num_conv = self.heads[head]
 
  conv_layers = []
+ c_in = in_channels
  for i in range(num_conv - 1):
  conv_layers.append(
  ConvModule(
- in_channels,
+ c_in,
  head_conv,
  kernel_size=final_kernel,
  stride=1,
  padding=final_kernel // 2,
  bias=bias,
  conv_cfg=conv_cfg,
  norm_cfg=norm_cfg))
+ c_in = head_conv
 
  conv_layers.append(
  build_conv_layer(