update backend

This commit is contained in:
2024-09-23 15:33:12 +08:00
parent 7c713a9c4c
commit 16d1f8ab67
3 changed files with 131 additions and 23 deletions

8
pts.py
View File

@@ -1,5 +1,6 @@
import numpy as np
import open3d as o3d
import torch
class PtsUtil:
@@ -18,5 +19,10 @@ class PtsUtil:
@staticmethod
def random_downsample_point_cloud(point_cloud, num_points):
idx = np.random.choice(len(point_cloud), num_points, replace=False)
idx = np.random.choice(len(point_cloud), num_points, replace=True)
return point_cloud[idx]
@staticmethod
def random_downsample_point_cloud_tensor(point_cloud, num_points):
idx = torch.randint(0, len(point_cloud), (num_points,))
return point_cloud[idx]