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
OpenPCDet
Commits
d35088b6
Commit
d35088b6
authored
Aug 20, 2022
by
Shaoshuai Shi
Browse files
check np.float32/np.float64 when using create_integrated_database.py
parent
96ba76a3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
0 deletions
+6
-0
tools/process_tools/create_integrated_databse.py
tools/process_tools/create_integrated_databse.py
+6
-0
No files found.
tools/process_tools/create_integrated_databse.py
View file @
d35088b6
...
@@ -29,6 +29,12 @@ def create_integrated_db_with_infos(args, root_path):
...
@@ -29,6 +29,12 @@ def create_integrated_db_with_infos(args, root_path):
obj_points
=
np
.
fromfile
(
str
(
obj_path
),
dtype
=
np
.
float32
).
reshape
(
obj_points
=
np
.
fromfile
(
str
(
obj_path
),
dtype
=
np
.
float32
).
reshape
(
[
-
1
,
args
.
num_point_features
])
[
-
1
,
args
.
num_point_features
])
num_points
=
obj_points
.
shape
[
0
]
num_points
=
obj_points
.
shape
[
0
]
if
num_points
!=
info
[
'num_points_in_gt'
]:
obj_points
=
np
.
fromfile
(
str
(
obj_path
),
dtype
=
np
.
float64
).
reshape
([
-
1
,
args
.
num_point_features
])
num_points
=
obj_points
.
shape
[
0
]
obj_points
=
obj_points
.
astype
(
np
.
float32
)
assert
num_points
==
info
[
'num_points_in_gt'
]
db_info_global
[
category
][
idx
][
'global_data_offset'
]
=
(
start_idx
,
start_idx
+
num_points
)
db_info_global
[
category
][
idx
][
'global_data_offset'
]
=
(
start_idx
,
start_idx
+
num_points
)
start_idx
+=
num_points
start_idx
+=
num_points
global_db_list
.
append
(
obj_points
)
global_db_list
.
append
(
obj_points
)
...
...
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