Upload app.py

app.py

@@ -5,6 +5,14 @@ import numpy
 import trimesh
 
 
+def random_sampling(points, num_samples):
+    # 如果点云数量小于采样数量，允许重复选择点
+    if len(points) < num_samples:
+        indices = numpy.random.choice(len(points), num_samples, replace=True)
+    else:
+        indices = numpy.random.choice(len(points), num_samples, replace=False)
+    sampled_points = points[indices]
+    return sampled_points
 
 def model_to_pc(mesh: trimesh.Trimesh, n_sample_points=10000):
     f32 = numpy.float32
@@ -99,6 +107,7 @@ for objaid in random_object_uids:
     objamodel = objaverse.load_objects([objaid])[objaid]
     try:
         pc = trimesh_to_pc(trimesh.load(objamodel))
+        pc = random_sampling(pc, num_samples=5000)
         pc_list.append(pc.reshape([1, -1, pc.shape[-1]]))
     except:
         continue