Spaces:
Build error
Build error
switch gl to cuda
Browse files- mesh_reconstruction/render.py +2 -2
- scripts/project_mesh.py +1 -1
mesh_reconstruction/render.py
CHANGED
|
@@ -14,7 +14,7 @@ def _warmup(glctx, device=None):
|
|
| 14 |
|
| 15 |
class NormalsRenderer:
|
| 16 |
|
| 17 |
-
_glctx:dr.
|
| 18 |
|
| 19 |
def __init__(
|
| 20 |
self,
|
|
@@ -29,7 +29,7 @@ class NormalsRenderer:
|
|
| 29 |
else:
|
| 30 |
self._mvp = mvp
|
| 31 |
self._image_size = image_size
|
| 32 |
-
self._glctx = dr.
|
| 33 |
_warmup(self._glctx, device)
|
| 34 |
|
| 35 |
def render(self,
|
|
|
|
| 14 |
|
| 15 |
class NormalsRenderer:
|
| 16 |
|
| 17 |
+
_glctx:dr.RasterizeCudaContext = None
|
| 18 |
|
| 19 |
def __init__(
|
| 20 |
self,
|
|
|
|
| 29 |
else:
|
| 30 |
self._mvp = mvp
|
| 31 |
self._image_size = image_size
|
| 32 |
+
self._glctx = dr.RasterizeCudaContext(output_db=False, 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.
|
| 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(output_db=False, device=device)
|
| 60 |
self.device = device
|
| 61 |
_warmup(self._glctx, device)
|
| 62 |
|