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
5592a505
Unverified
Commit
5592a505
authored
Feb 17, 2021
by
Yezhen Cong
Committed by
GitHub
Feb 17, 2021
Browse files
[Fix] Fix typos and error in comments (#308)
* fix bug caused by mmcv upgrade; delete pdb breakpoint * fix typos
parent
ac9a3e8c
Changes
14
Hide whitespace changes
Inline
Side-by-side
Showing
14 changed files
with
30 additions
and
30 deletions
+30
-30
configs/_base_/models/centerpoint_01voxel_second_secfpn_nus.py
...gs/_base_/models/centerpoint_01voxel_second_secfpn_nus.py
+1
-1
configs/_base_/models/centerpoint_02pillar_second_secfpn_nus.py
...s/_base_/models/centerpoint_02pillar_second_secfpn_nus.py
+1
-1
configs/centerpoint/centerpoint_0075voxel_second_secfpn_dcn_4x8_cyclic_20e_nus.py
...erpoint_0075voxel_second_secfpn_dcn_4x8_cyclic_20e_nus.py
+2
-2
configs/centerpoint/centerpoint_0075voxel_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py
...75voxel_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py
+2
-2
configs/centerpoint/centerpoint_01voxel_second_secfpn_dcn_4x8_cyclic_20e_nus.py
...nterpoint_01voxel_second_secfpn_dcn_4x8_cyclic_20e_nus.py
+2
-2
configs/centerpoint/centerpoint_01voxel_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py
...01voxel_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py
+2
-2
configs/centerpoint/centerpoint_02pillar_second_secfpn_dcn_4x8_cyclic_20e_nus.py
...terpoint_02pillar_second_secfpn_dcn_4x8_cyclic_20e_nus.py
+2
-2
configs/centerpoint/centerpoint_02pillar_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py
...2pillar_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py
+2
-2
mmdet3d/models/dense_heads/centerpoint_head.py
mmdet3d/models/dense_heads/centerpoint_head.py
+7
-7
mmdet3d/models/fusion_layers/point_fusion.py
mmdet3d/models/fusion_layers/point_fusion.py
+1
-1
tests/test_models/test_forward.py
tests/test_models/test_forward.py
+2
-2
tests/test_models/test_heads/test_heads.py
tests/test_models/test_heads/test_heads.py
+2
-2
tests/test_utils/test_anchors.py
tests/test_utils/test_anchors.py
+2
-2
tests/test_utils/test_assigners.py
tests/test_utils/test_assigners.py
+2
-2
No files found.
configs/_base_/models/centerpoint_01voxel_second_secfpn_nus.py
View file @
5592a505
...
...
@@ -52,7 +52,7 @@ model = dict(
out_size_factor
=
8
,
voxel_size
=
voxel_size
[:
2
],
code_size
=
9
),
sep
e
rate_head
=
dict
(
sep
a
rate_head
=
dict
(
type
=
'SeparateHead'
,
init_bias
=-
2.19
,
final_kernel
=
3
),
loss_cls
=
dict
(
type
=
'GaussianFocalLoss'
,
reduction
=
'mean'
),
loss_bbox
=
dict
(
type
=
'L1Loss'
,
reduction
=
'mean'
,
loss_weight
=
0.25
),
...
...
configs/_base_/models/centerpoint_02pillar_second_secfpn_nus.py
View file @
5592a505
...
...
@@ -51,7 +51,7 @@ model = dict(
out_size_factor
=
4
,
voxel_size
=
voxel_size
[:
2
],
code_size
=
9
),
sep
e
rate_head
=
dict
(
sep
a
rate_head
=
dict
(
type
=
'SeparateHead'
,
init_bias
=-
2.19
,
final_kernel
=
3
),
loss_cls
=
dict
(
type
=
'GaussianFocalLoss'
,
reduction
=
'mean'
),
loss_bbox
=
dict
(
type
=
'L1Loss'
,
reduction
=
'mean'
,
loss_weight
=
0.25
),
...
...
configs/centerpoint/centerpoint_0075voxel_second_secfpn_dcn_4x8_cyclic_20e_nus.py
View file @
5592a505
...
...
@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_0075voxel_second_secfpn_4x8_cyclic_20e_nus.py']
model
=
dict
(
pts_bbox_head
=
dict
(
sep
e
rate_head
=
dict
(
type
=
'DCNSep
e
rateHead'
,
sep
a
rate_head
=
dict
(
type
=
'DCNSep
a
rateHead'
,
dcn_config
=
dict
(
type
=
'DCN'
,
in_channels
=
64
,
...
...
configs/centerpoint/centerpoint_0075voxel_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py
View file @
5592a505
...
...
@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_0075voxel_second_secfpn_4x8_cyclic_20e_nus.py']
model
=
dict
(
pts_bbox_head
=
dict
(
sep
e
rate_head
=
dict
(
type
=
'DCNSep
e
rateHead'
,
sep
a
rate_head
=
dict
(
type
=
'DCNSep
a
rateHead'
,
dcn_config
=
dict
(
type
=
'DCN'
,
in_channels
=
64
,
...
...
configs/centerpoint/centerpoint_01voxel_second_secfpn_dcn_4x8_cyclic_20e_nus.py
View file @
5592a505
...
...
@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_01voxel_second_secfpn_4x8_cyclic_20e_nus.py']
model
=
dict
(
pts_bbox_head
=
dict
(
sep
e
rate_head
=
dict
(
type
=
'DCNSep
e
rateHead'
,
sep
a
rate_head
=
dict
(
type
=
'DCNSep
a
rateHead'
,
dcn_config
=
dict
(
type
=
'DCN'
,
in_channels
=
64
,
...
...
configs/centerpoint/centerpoint_01voxel_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py
View file @
5592a505
...
...
@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_01voxel_second_secfpn_4x8_cyclic_20e_nus.py']
model
=
dict
(
pts_bbox_head
=
dict
(
sep
e
rate_head
=
dict
(
type
=
'DCNSep
e
rateHead'
,
sep
a
rate_head
=
dict
(
type
=
'DCNSep
a
rateHead'
,
dcn_config
=
dict
(
type
=
'DCN'
,
in_channels
=
64
,
...
...
configs/centerpoint/centerpoint_02pillar_second_secfpn_dcn_4x8_cyclic_20e_nus.py
View file @
5592a505
...
...
@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py']
model
=
dict
(
pts_bbox_head
=
dict
(
sep
e
rate_head
=
dict
(
type
=
'DCNSep
e
rateHead'
,
sep
a
rate_head
=
dict
(
type
=
'DCNSep
a
rateHead'
,
dcn_config
=
dict
(
type
=
'DCN'
,
in_channels
=
64
,
...
...
configs/centerpoint/centerpoint_02pillar_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py
View file @
5592a505
...
...
@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py']
model
=
dict
(
pts_bbox_head
=
dict
(
sep
e
rate_head
=
dict
(
type
=
'DCNSep
e
rateHead'
,
sep
a
rate_head
=
dict
(
type
=
'DCNSep
a
rateHead'
,
dcn_config
=
dict
(
type
=
'DCN'
,
in_channels
=
64
,
...
...
mmdet3d/models/dense_heads/centerpoint_head.py
View file @
5592a505
...
...
@@ -119,8 +119,8 @@ class SeparateHead(nn.Module):
@
HEADS
.
register_module
()
class
DCNSep
e
rateHead
(
nn
.
Module
):
r
"""DCNSep
e
rateHead for CenterHead.
class
DCNSep
a
rateHead
(
nn
.
Module
):
r
"""DCNSep
a
rateHead for CenterHead.
.. code-block:: none
/-----> DCN for heatmap task -----> heatmap task.
...
...
@@ -155,7 +155,7 @@ class DCNSeperateHead(nn.Module):
norm_cfg
=
dict
(
type
=
'BN2d'
),
bias
=
'auto'
,
**
kwargs
):
super
(
DCNSep
e
rateHead
,
self
).
__init__
()
super
(
DCNSep
a
rateHead
,
self
).
__init__
()
if
'heatmap'
in
heads
:
heads
.
pop
(
'heatmap'
)
# feature adaptation with dcn
...
...
@@ -250,7 +250,7 @@ class CenterHead(nn.Module):
Default: dict(type='GaussianFocalLoss', reduction='mean').
loss_bbox (dict): Config of regression loss function.
Default: dict(type='L1Loss', reduction='none').
sep
e
rate_head (dict): Config of sep
e
rate head. Default: dict(
sep
a
rate_head (dict): Config of sep
a
rate head. Default: dict(
type='SeparateHead', init_bias=-2.19, final_kernel=3)
share_conv_channel (int): Output channels for share_conv_layer.
Default: 64.
...
...
@@ -273,7 +273,7 @@ class CenterHead(nn.Module):
loss_cls
=
dict
(
type
=
'GaussianFocalLoss'
,
reduction
=
'mean'
),
loss_bbox
=
dict
(
type
=
'L1Loss'
,
reduction
=
'none'
,
loss_weight
=
0.25
),
sep
e
rate_head
=
dict
(
sep
a
rate_head
=
dict
(
type
=
'SeparateHead'
,
init_bias
=-
2.19
,
final_kernel
=
3
),
share_conv_channel
=
64
,
num_heatmap_convs
=
2
,
...
...
@@ -312,9 +312,9 @@ class CenterHead(nn.Module):
for
num_cls
in
num_classes
:
heads
=
copy
.
deepcopy
(
common_heads
)
heads
.
update
(
dict
(
heatmap
=
(
num_cls
,
num_heatmap_convs
)))
sep
e
rate_head
.
update
(
sep
a
rate_head
.
update
(
in_channels
=
share_conv_channel
,
heads
=
heads
,
num_cls
=
num_cls
)
self
.
task_heads
.
append
(
builder
.
build_head
(
sep
e
rate_head
))
self
.
task_heads
.
append
(
builder
.
build_head
(
sep
a
rate_head
))
def
init_weights
(
self
):
"""Initialize weights."""
...
...
mmdet3d/models/fusion_layers/point_fusion.py
View file @
5592a505
...
...
@@ -291,7 +291,7 @@ class PointFusion(nn.Module):
Args:
img_feats (torch.Tensor): Image feature map in shape
(
N
, C, H, W).
(
1
, C, H, W).
pts (torch.Tensor): Points of a single sample.
img_meta (dict): Meta information of the single sample.
...
...
tests/test_models/test_forward.py
View file @
5592a505
"""Test model forward process.
CommandLine:
pytest tests/test_forward.py
xdoctest tests/test_forward.py zero
pytest tests/test_
models/test_
forward.py
xdoctest tests/test_
models/test_
forward.py zero
"""
import
copy
import
numpy
as
np
...
...
tests/test_models/test_heads/test_heads.py
View file @
5592a505
...
...
@@ -806,8 +806,8 @@ def test_dcn_center_head():
out_size_factor
=
4
,
voxel_size
=
voxel_size
[:
2
],
code_size
=
9
),
sep
e
rate_head
=
dict
(
type
=
'DCNSep
e
rateHead'
,
sep
a
rate_head
=
dict
(
type
=
'DCNSep
a
rateHead'
,
dcn_config
=
dict
(
type
=
'DCN'
,
in_channels
=
64
,
...
...
tests/test_utils/test_anchors.py
View file @
5592a505
"""
CommandLine:
pytest tests/test_anchor.py
xdoctest tests/test_anchor.py zero
pytest tests/test_
utils/test_
anchor.py
xdoctest tests/test_
utils/test_
anchor.py zero
"""
import
torch
...
...
tests/test_utils/test_assigners.py
View file @
5592a505
"""Tests the Assigner objects.
CommandLine:
pytest tests/test_assigner.py
xdoctest tests/test_assigner.py zero
pytest tests/test_
utils/test_
assigner.py
xdoctest tests/test_
utils/test_
assigner.py zero
"""
import
torch
...
...
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