Update models/ISOMER/mesh_reconstruction/render.py
Browse files
models/ISOMER/mesh_reconstruction/render.py
CHANGED
@@ -45,7 +45,10 @@ class NormalsRenderer:
|
|
45 |
else:
|
46 |
self._mvp = mvp
|
47 |
self._image_size = image_size
|
48 |
-
|
|
|
|
|
|
|
49 |
_warmup(self._glctx, device)
|
50 |
|
51 |
def render(self,
|
|
|
45 |
else:
|
46 |
self._mvp = mvp
|
47 |
self._image_size = image_size
|
48 |
+
if glctx==None:
|
49 |
+
self._glctx = dr.RasterizeCudaContext(device="cuda")
|
50 |
+
else:
|
51 |
+
self._glctx = glctx
|
52 |
_warmup(self._glctx, device)
|
53 |
|
54 |
def render(self,
|