add

.gitignore

@@ -1,9 +1,9 @@
-data/
 checkpoints/
 venv/
 log_feature/
 log/
-trex/
 log_style/
+data/
+trex/
 
 

requirements.txt

@@ -2,7 +2,6 @@ pytorch_lightning==1.9.0
 transformers==4.26.1
 pytorch_metric_learning==2.0.1
 pandas==1.5.3
-pymeshlab
 tabulate
 torchvision
 matplotlib

src/utils/loading.py

@@ -1,26 +1,6 @@
-import random
-import numpy as np
-import pymeshlab
 import yaml
 
 
 def load_yaml(path):
     with open(path, "rt") as f:
         return yaml.safe_load(f)
-
-
-def load_point_cloud(filename: str, n_sample: int = 1024):
-    ms = pymeshlab.MeshSet()
-    ms.load_new_mesh(filename)
-    ms.generate_sampling_poisson_disk(samplenum=n_sample)
-    pc_list = ms.current_mesh().vertex_matrix()
-
-    n = len(pc_list)
-    if n > n_sample:
-        pc_list = np.array(random.choices(pc_list, k=n_sample))
-    elif n < n_sample:
-        pc_list = np.concatenate(
-            [pc_list, np.array(random.choices(pc_list, k=n_sample - n))], axis=0
-        )
-
-    return pc_list