Skip to content

GitLab

Unverified Commit bafee2ec authored by Zhe Zhao's avatar Zhe Zhao Committed by GitHub
Browse files

Fix channel bug in centerpoint Separatehead (#228)

parent e5a155e8
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
mmdet3d/models/dense_heads/centerpoint_head.py
View file @ bafee2ec
...@@ -51,10 +51,11 @@ class SeparateHead(nn.Module): ...@@ -51,10 +51,11 @@ class SeparateHead(nn.Module):
classes, num_conv = self.heads[head] classes, num_conv = self.heads[head]
conv_layers = [] conv_layers = []
c_in = in_channels
for i in range(num_conv - 1): for i in range(num_conv - 1):
conv_layers.append( conv_layers.append(
ConvModule( ConvModule(
in_channels, c_in,
head_conv, head_conv,
kernel_size=final_kernel, kernel_size=final_kernel,
stride=1, stride=1,
...@@ -62,6 +63,7 @@ class SeparateHead(nn.Module): ...@@ -62,6 +63,7 @@ class SeparateHead(nn.Module):
bias=bias, bias=bias,
conv_cfg=conv_cfg, conv_cfg=conv_cfg,
norm_cfg=norm_cfg)) norm_cfg=norm_cfg))
c_in = head_conv
conv_layers.append( conv_layers.append(
build_conv_layer( build_conv_layer(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment