- mesh_reconstruction/render.py +1 -1
- scripts/project_mesh.py +1 -1
mesh_reconstruction/render.py
CHANGED
@@ -29,7 +29,7 @@ class NormalsRenderer:
|
|
29 |
else:
|
30 |
self._mvp = mvp
|
31 |
self._image_size = image_size
|
32 |
-
self._glctx = dr.RasterizeCudaContext(
|
33 |
_warmup(self._glctx, device)
|
34 |
|
35 |
def render(self,
|
|
|
29 |
else:
|
30 |
self._mvp = mvp
|
31 |
self._image_size = image_size
|
32 |
+
self._glctx = dr.RasterizeCudaContext(device=device)
|
33 |
_warmup(self._glctx, device)
|
34 |
|
35 |
def render(self,
|
scripts/project_mesh.py
CHANGED
@@ -56,7 +56,7 @@ def _warmup(glctx, device=None):
|
|
56 |
|
57 |
class Pix2FacesRenderer:
|
58 |
def __init__(self, device="cuda"):
|
59 |
-
self._glctx = dr.RasterizeCudaContext(
|
60 |
self.device = device
|
61 |
_warmup(self._glctx, device)
|
62 |
|
|
|
56 |
|
57 |
class Pix2FacesRenderer:
|
58 |
def __init__(self, device="cuda"):
|
59 |
+
self._glctx = dr.RasterizeCudaContext(device=device)
|
60 |
self.device = device
|
61 |
_warmup(self._glctx, device)
|
62 |
|