6D prediction files now working

2020-06-25 12:03:08 +02:00
parent 965b805ee9
commit 2a7a236b89
5 changed files with 113 additions and 80 deletions
@@ -75,9 +75,12 @@ def write_pointcloud(file, pc, numCols=6):


 def farthest_point_sampling(pts, K):
-    if isinstance(pts, Data):
-        pts = pts.pos.numpy()
-    if pts.shape[0] < K:
+    if K > 0:
+        if isinstance(pts, Data):
+            pts = pts.pos.numpy()
+        if pts.shape[0] < K:
+            return pts
+    else:
        return pts

    def calc_distances(p0, points):