update

primitive_anything/primitive_transformer.py

@@ -11,6 +11,7 @@ from einops import rearrange, repeat, reduce, pack
 from gateloop_transformer import SimpleGateLoopLayer
 from huggingface_hub import PyTorchModelHubMixin
 import numpy as np
+import trimesh
 from tqdm import tqdm
 import torch
 from torch import nn, Tensor
@@ -322,8 +323,8 @@ class PrimitiveTransformerDiscrete(Module, PyTorchModelHubMixin):
 
         bs_pc_map = {}
         for bs_name, type_code in SHAPE_CODE.items():
-            pc = o3d.io.read_point_cloud(os.path.join(bs_pc_dir, f'SM_GR_BS_{bs_name}_001.ply'))
-            bs_pc_map[type_code] = torch.from_numpy(np.asarray(pc.points)).float()
+            pc = trimesh.load(os.path.join(bs_pc_dir, f'SM_GR_BS_{bs_name}_001.ply'))
+            bs_pc_map[type_code] = torch.from_numpy(np.asarray(pc.vertices)).float()
         bs_pc_list = []
         for i in range(len(bs_pc_map)):
             bs_pc_list.append(bs_pc_map[i])