Commit 792473c9 authored by MeowZheng's avatar MeowZheng Committed by Wenwei Zhang
Browse files

return2returns

parent b0582217
...@@ -71,7 +71,7 @@ class QueryAndGroup(nn.Module): ...@@ -71,7 +71,7 @@ class QueryAndGroup(nn.Module):
center_xyz (Tensor): (B, npoint, 3) coordinates of the centriods. center_xyz (Tensor): (B, npoint, 3) coordinates of the centriods.
features (Tensor): (B, C, N) Descriptors of the features. features (Tensor): (B, C, N) Descriptors of the features.
Return: Returns:
Tensor: (B, 3 + C, npoint, sample_num) Grouped feature. Tensor: (B, 3 + C, npoint, sample_num) Grouped feature.
""" """
# if self.max_radius is None, we will perform kNN instead of ball query # if self.max_radius is None, we will perform kNN instead of ball query
...@@ -152,7 +152,7 @@ class GroupAll(nn.Module): ...@@ -152,7 +152,7 @@ class GroupAll(nn.Module):
new_xyz (Tensor): new xyz coordinates of the features. new_xyz (Tensor): new xyz coordinates of the features.
features (Tensor): (B, C, N) features to group. features (Tensor): (B, C, N) features to group.
Return: Returns:
Tensor: (B, C + 3, 1, N) Grouped feature. Tensor: (B, C + 3, 1, N) Grouped feature.
""" """
grouped_xyz = xyz.transpose(1, 2).unsqueeze(2) grouped_xyz = xyz.transpose(1, 2).unsqueeze(2)
......
...@@ -95,7 +95,7 @@ class PointsSampler(nn.Module): ...@@ -95,7 +95,7 @@ class PointsSampler(nn.Module):
points_xyz (Tensor): (B, N, 3) xyz coordinates of the features. points_xyz (Tensor): (B, N, 3) xyz coordinates of the features.
features (Tensor): (B, C, N) Descriptors of the features. features (Tensor): (B, C, N) Descriptors of the features.
Return: Returns:
Tensor: (B, npoint, sample_num) Indices of sampled points. Tensor: (B, npoint, sample_num) Indices of sampled points.
""" """
indices = [] indices = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment