update

2024-10-26 03:24:01 -05:00
parent a1226eb294
commit a41571e79c
6 changed files with 121 additions and 18 deletions
--- a/utils/data_load.py
+++ b/utils/data_load.py
@@ -210,6 +210,17 @@ class DataLoadUtil:
        else:
            pts = np.load(npy_path)
        return pts
+    
+    @staticmethod
+    def load_from_preprocessed_nrm(path, file_type="npy"):
+        npy_path = os.path.join(
+            os.path.dirname(path), "nrm", os.path.basename(path) + "." + file_type
+        )
+        if file_type == "txt":
+            nrm = np.loadtxt(npy_path)
+        else:
+            nrm = np.load(npy_path)
+        return nrm

    @staticmethod
    def cam_pose_transformation(cam_pose_before):