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
aef0bfd1
Unverified
Commit
aef0bfd1
authored
Sep 16, 2022
by
VVsssssk
Committed by
GitHub
Sep 16, 2022
Browse files
[Fix]Fix dev-1.x branch ci (#1813)
* add deploy.yaml * fix ut * fix update data script description
parent
14ccfc43
Changes
6
Hide whitespace changes
Inline
Side-by-side
Showing
6 changed files
with
14 additions
and
13 deletions
+14
-13
configs/3dssd/3dssd_4xb4_kitti-3d-car.py
configs/3dssd/3dssd_4xb4_kitti-3d-car.py
+0
-0
tests/test_datasets/test_kitti_dataset.py
tests/test_datasets/test_kitti_dataset.py
+3
-3
tests/test_datasets/test_lyft_dataset.py
tests/test_datasets/test_lyft_dataset.py
+4
-4
tests/test_datasets/test_nuscenes_dataset.py
tests/test_datasets/test_nuscenes_dataset.py
+4
-4
tests/test_models/test_dense_heads/test_freeanchors.py
tests/test_models/test_dense_heads/test_freeanchors.py
+2
-2
tools/dataset_converters/update_infos_to_v2.py
tools/dataset_converters/update_infos_to_v2.py
+1
-0
No files found.
configs/3dssd/3dssd_4xb4_kitti-3d-car.py
.py
→
configs/3dssd/3dssd_4xb4_kitti-3d-car.py
View file @
aef0bfd1
File moved
tests/test_datasets/test_kitti_dataset.py
View file @
aef0bfd1
...
...
@@ -28,7 +28,7 @@ def _generate_kitti_dataset_config():
gt_instances_3d
=
InstanceData
()
gt_instances_3d
.
labels_3d
=
info
[
'gt_labels_3d'
]
data_sample
.
gt_instances_3d
=
gt_instances_3d
info
[
'data_sample'
]
=
data_sample
info
[
'data_sample
s
'
]
=
data_sample
return
info
pipeline
=
[
...
...
@@ -82,9 +82,9 @@ def test_getitem():
assert
torch
.
allclose
(
ann_info
[
'gt_bboxes_3d'
].
tensor
.
sum
(),
torch
.
tensor
(
7.2650
))
assert
'centers_2d'
in
ann_info
assert
ann_info
[
'centers_2d'
].
dtype
==
np
.
float
64
assert
ann_info
[
'centers_2d'
].
dtype
==
np
.
float
32
assert
'depths'
in
ann_info
assert
ann_info
[
'depths'
].
dtype
==
np
.
float
64
assert
ann_info
[
'depths'
].
dtype
==
np
.
float
32
car_kitti_dataset
=
KittiDataset
(
data_root
,
...
...
tests/test_datasets/test_lyft_dataset.py
View file @
aef0bfd1
...
...
@@ -21,12 +21,12 @@ def _generate_nus_dataset_config():
class
Identity
(
BaseTransform
):
def
transform
(
self
,
info
):
packed_input
=
dict
(
data_sample
=
Det3DDataSample
())
packed_input
=
dict
(
data_sample
s
=
Det3DDataSample
())
if
'ann_info'
in
info
:
packed_input
[
'data_sample'
].
gt_instances_3d
=
InstanceData
(
)
packed_input
[
'data_sample'
].
gt_instances_3d
.
labels_3d
=
info
[
'data_samples'
].
gt_instances_3d
=
InstanceData
()
packed_input
[
'data_samples'
].
gt_instances_3d
.
labels_3d
=
info
[
'ann_info'
][
'gt_labels_3d'
]
return
packed_input
...
...
tests/test_datasets/test_nuscenes_dataset.py
View file @
aef0bfd1
...
...
@@ -21,12 +21,12 @@ def _generate_nus_dataset_config():
class
Identity
(
BaseTransform
):
def
transform
(
self
,
info
):
packed_input
=
dict
(
data_sample
=
Det3DDataSample
())
packed_input
=
dict
(
data_sample
s
=
Det3DDataSample
())
if
'ann_info'
in
info
:
packed_input
[
'data_sample'
].
gt_instances_3d
=
InstanceData
(
)
packed_input
[
'data_sample'
].
gt_instances_3d
.
labels_3d
=
info
[
'data_samples'
].
gt_instances_3d
=
InstanceData
()
packed_input
[
'data_samples'
].
gt_instances_3d
.
labels_3d
=
info
[
'ann_info'
][
'gt_labels_3d'
]
return
packed_input
...
...
tests/test_models/test_dense_heads/test_freeanchors.py
View file @
aef0bfd1
...
...
@@ -17,8 +17,8 @@ class TestFreeAnchor(unittest.TestCase):
DefaultScope
.
get_instance
(
'test_freeanchor'
,
scope_name
=
'mmdet3d'
)
_setup_seed
(
0
)
freeanchor_cfg
=
_get_detector_cfg
(
'free_anchor/
hv_
pointpillars_
fpn_sbn-all_free-
'
'
anchor_4x8_
2x_nus-3d.py'
)
'free_anchor/pointpillars_
hv_regnet-1.6gf_fpn_head-free-anchor
'
'
_sbn-all_8xb4-
2x_nus-3d.py'
)
model
=
MODELS
.
build
(
freeanchor_cfg
)
num_gt_instance
=
3
packed_inputs
=
_create_detector_inputs
(
...
...
tools/dataset_converters/update_infos_to_v2.py
View file @
aef0bfd1
...
...
@@ -3,6 +3,7 @@
Example:
python tools/dataset_converters/update_infos_to_v2.py
--dataset kitti
--pkl ./data/kitti/kitti_infos_train.pkl
--out-dir ./kitti_v2/
"""
...
...
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