Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
mmdetection3d
Commits
bafee2ec
"git@developer.sourcefind.cn:OpenDAS/d2go.git" did not exist on "70f157c5bd01f62fc1662e8d8bd8593a50bc4c4d"
Unverified
Commit
bafee2ec
authored
Dec 13, 2020
by
Zhe Zhao
Committed by
GitHub
Dec 13, 2020
Browse files
Fix channel bug in centerpoint Separatehead (#228)
parent
e5a155e8
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
mmdet3d/models/dense_heads/centerpoint_head.py
mmdet3d/models/dense_heads/centerpoint_head.py
+3
-1
No files found.
mmdet3d/models/dense_heads/centerpoint_head.py
View file @
bafee2ec
...
...
@@ -51,10 +51,11 @@ class SeparateHead(nn.Module):
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
,
...
...
@@ -62,6 +63,7 @@ class SeparateHead(nn.Module):
bias
=
bias
,
conv_cfg
=
conv_cfg
,
norm_cfg
=
norm_cfg
))
c_in
=
head_conv
conv_layers
.
append
(
build_conv_layer
(
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment