Added normals to prediction DataObject

2019-08-09 12:35:55 +02:00
parent 8eb165f76c
commit 39e5d72226
3 changed files with 21 additions and 22 deletions
--- a/model/pointnet2_part_seg.py
+++ b/model/pointnet2_part_seg.py
@@ -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 = 6
+GLOBAL_POINT_FEATURES = 3

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