LTT commited on
Commit
bca10b4
Β·
verified Β·
1 Parent(s): cdb6265

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
- self._glctx = glctx
 
 
 
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,