Added normals to prediction DataObject

2019-08-09 13:06:29 +02:00
parent 39e5d72226
commit a501dcd6b0
2 changed files with 8 additions and 8 deletions
@@ -8,7 +8,7 @@ from torch_geometric.utils.num_nodes import maybe_num_nodes
 from torch_geometric.data.data import Data
 from torch_scatter import scatter_add, scatter_max

-GLOBAL_POINT_FEATURES = 3
+GLOBAL_POINT_FEATURES = 6

 class PointNet2SAModule(torch.nn.Module):
    def __init__(self, sample_radio, radius, max_num_neighbors, mlp):