Added normals to prediction DataObject

2019-08-09 10:25:16 +02:00
parent 4e1fcdfd43
commit 8eb165f76c
7 changed files with 101464 additions and 2060 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 = 3
+GLOBAL_POINT_FEATURES = 6

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