try osemesa

8d35c806 · Low, Low Jian He (PG/R - Comp Sci & Elec Eng) · 683c077f · 8d35c806
Commit 8d35c806 authored 1 month ago by Low, Low Jian He (PG/R - Comp Sci & Elec Eng)
--- a/hamer/utils/mesh_renderer.py
+++ b/hamer/utils/mesh_renderer.py
 import os
+# if 'PYOPENGL_PLATFORM' not in os.environ:
+#     os.environ['PYOPENGL_PLATFORM'] = 'egl'
 if 'PYOPENGL_PLATFORM' not in os.environ:
-    os.environ['PYOPENGL_PLATFORM'] = 'egl'
+    os.environ['PYOPENGL_PLATFORM'] = 'osmesa'
 import torch
 from torchvision.utils import make_grid
 import numpy as np