solve merge

2024-10-28 18:27:16 +00:00
parent 3c9e2c8d12 a883a31968
commit 9e39c6c6c9
9 changed files with 237 additions and 19 deletions
--- a/utils/data_load.py
+++ b/utils/data_load.py
@@ -212,6 +212,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):