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
31165186
Unverified
Commit
31165186
authored
Apr 08, 2021
by
Ziyi Wu
Committed by
GitHub
Apr 08, 2021
Browse files
[Fix]: fix point type errors in function docstring (#422)
parent
ecd0d06a
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
13 additions
and
12 deletions
+13
-12
mmdet3d/datasets/pipelines/loading.py
mmdet3d/datasets/pipelines/loading.py
+5
-4
mmdet3d/datasets/pipelines/transforms_3d.py
mmdet3d/datasets/pipelines/transforms_3d.py
+8
-8
No files found.
mmdet3d/datasets/pipelines/loading.py
View file @
31165186
...
...
@@ -154,7 +154,7 @@ class LoadPointsFromMultiSweeps(object):
"""Removes point too close within a certain radius from origin.
Args:
points (np.ndarray): Sweep points.
points (np.ndarray
| :obj:`BasePoints`
): Sweep points.
radius (float): Radius below which points are removed.
Defaults to 1.0.
...
...
@@ -183,7 +183,8 @@ class LoadPointsFromMultiSweeps(object):
dict: The result dict containing the multi-sweep points data.
\
Added key and value are described below.
- points (np.ndarray): Multi-sweep point cloud arrays.
- points (np.ndarray | :obj:`BasePoints`): Multi-sweep point
\
cloud arrays.
"""
points
=
results
[
'points'
]
points
.
tensor
[:,
4
]
=
0
...
...
@@ -295,7 +296,7 @@ class NormalizePointsColor(object):
dict: The result dict containing the normalized points.
\
Updated key and value are described below.
- points (
np.ndarray
): Points after color normalization.
- points (
:obj:`BasePoints`
): Points after color normalization.
"""
points
=
results
[
'points'
]
assert
points
.
attribute_dims
is
not
None
and
\
...
...
@@ -393,7 +394,7 @@ class LoadPointsFromFile(object):
dict: The result dict containing the point clouds data.
\
Added key and value are described below.
- points (
np.ndarray
): Point clouds data.
- points (
:obj:`BasePoints`
): Point clouds data.
"""
pts_filename
=
results
[
'pts_filename'
]
points
=
self
.
_load_points
(
pts_filename
)
...
...
mmdet3d/datasets/pipelines/transforms_3d.py
View file @
31165186
...
...
@@ -147,7 +147,7 @@ class ObjectSample(object):
"""Remove the points in the sampled bounding boxes.
Args:
points (
np.ndarray
): Input point cloud array.
points (
:obj:`BasePoints`
): Input point cloud array.
boxes (np.ndarray): Sampled ground truth boxes.
Returns:
...
...
@@ -604,7 +604,7 @@ class IndoorPointSample(object):
Sample points to a certain number.
Args:
points (np.ndarray): 3D Points.
points (np.ndarray
| :obj:`BasePoints`
): 3D Points.
num_samples (int): Number of samples to be sampled.
replace (bool): Whether the sample is with or without replacement.
Defaults to None.
...
...
@@ -613,7 +613,7 @@ class IndoorPointSample(object):
Returns:
tuple[np.ndarray] | np.ndarray:
- points (np.ndarray): 3D Points.
- points (np.ndarray
| :obj:`BasePoints`
): 3D Points.
- choices (np.ndarray, optional): The generated random samples.
"""
if
replace
is
None
:
...
...
@@ -708,10 +708,10 @@ class IndoorPatchPointSample(object):
attributes (np.ndarray): features of input points.
attribute_dims (dict): Dictionary to indicate the meaning of extra
dimension.
point_type (type): class of input points.
point_type (type): class of input points
inherited from BasePoints
.
Returns:
np.ndarray
: The generated input data.
:obj:`BasePoints`
: The generated input data.
"""
# subtract patch center, the z dimension is not centered
centered_coords
=
coords
.
copy
()
...
...
@@ -742,15 +742,15 @@ class IndoorPatchPointSample(object):
Then sample points within that patch to a certain number.
Args:
points (BasePoints): 3D Points.
points (
:obj:`
BasePoints
`
): 3D Points.
sem_mask (np.ndarray): semantic segmentation mask for input points.
replace (bool): Whether the sample is with or without replacement.
Defaults to None.
Returns:
tuple[
np.ndarray] | np.ndarray
:
tuple[
:obj:`BasePoints`, np.ndarray] | :obj:`BasePoints`
:
- points (BasePoints): 3D Points.
- points (
:obj:`
BasePoints
`
): 3D Points.
- choices (np.ndarray): The generated random samples.
"""
coords
=
points
.
coord
.
numpy
()
...
...
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