Skip to content

Commit

Permalink
[Doc]: Update description of Args: voxel_features (#1614)
Browse files Browse the repository at this point in the history
Motivation
Just better description in PointPillarScatter.

Modification
The original description of voxel_features is (N, M, C), which indeed is (N, C)

BC-breaking (Optional)
NO

Use cases (Optional)
NO

Checklist
Pre-commit or other linting tools are used to fix the potential lint issues.
The modification is covered by complete unit tests. If not, please add more unit test to ensure the correctness.
If the modification has potential influence on downstream projects, this PR should be tested with downstream projects.
The documentation has been modified accordingly, like docstring or example tutorials.
  • Loading branch information
ChonghaoSima authored Jul 19, 2022
1 parent f30aaec commit 0c7cab9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mmdet3d/models/middle_encoders/pillar_scatter.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ def forward_single(self, voxel_features, coors):
"""Scatter features of single sample.
Args:
voxel_features (torch.Tensor): Voxel features in shape (N, M, C).
voxel_features (torch.Tensor): Voxel features in shape (N, C).
coors (torch.Tensor): Coordinates of each voxel.
The first column indicates the sample ID.
"""
Expand All @@ -63,7 +63,7 @@ def forward_batch(self, voxel_features, coors, batch_size):
"""Scatter features of single sample.
Args:
voxel_features (torch.Tensor): Voxel features in shape (N, M, C).
voxel_features (torch.Tensor): Voxel features in shape (N, C).
coors (torch.Tensor): Coordinates of each voxel in shape (N, 4).
The first column indicates the sample ID.
batch_size (int): Number of samples in the current batch.
Expand Down

0 comments on commit 0c7cab9

Please sign in to comment.