Make pyrender optional for headless skeleton inference

skeleton/utils/save_utils.py

@@ -21,9 +21,15 @@ from collections import deque, defaultdict
 from scipy.cluster.hierarchy import linkage, fcluster
 from scipy.spatial.distance import cdist
 
-from data_utils.pyrender_wrapper import PyRenderWrapper
 from data_utils.data_loader import DataLoader
 
+_PYRENDER_IMPORT_ERROR = None
+try:
+    from data_utils.pyrender_wrapper import PyRenderWrapper
+except Exception as exc:
+    PyRenderWrapper = None
+    _PYRENDER_IMPORT_ERROR = exc
+
 def save_mesh(vertices, faces, filename):
 
     mesh = trimesh.Trimesh(vertices=vertices, faces=faces)    
@@ -597,6 +603,12 @@ def render_mesh_with_skeleton(joints, bones, vertices, faces, output_dir, filena
     """
     Render the mesh with skeleton using PyRender.
     """
+    if PyRenderWrapper is None:
+        raise RuntimeError(
+            "PyRender is unavailable in this runtime. "
+            f"Original import error: {_PYRENDER_IMPORT_ERROR}"
+        )
+
     loader = DataLoader()
     
     raw_size = (960, 960)
@@ -645,4 +657,4 @@ def save_args(args, output_dir, filename="config.json"):
     os.makedirs(output_dir, exist_ok=True)
     config_path = os.path.join(output_dir, filename)
     with open(config_path, 'w') as f:
-        json.dump(args_dict, f, indent=4)
+        json.dump(args_dict, f, indent=4)