Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
7c935af
1
Parent(s):
906626c
update
Browse files
common.py
CHANGED
@@ -377,8 +377,9 @@ def image_to_3d(
|
|
377 |
|
378 |
gs_model = outputs["gaussian"][0]
|
379 |
mesh_model = outputs["mesh"][0]
|
380 |
-
|
381 |
-
|
|
|
382 |
output_root = TMP_DIR
|
383 |
if req is not None:
|
384 |
output_root = os.path.join(output_root, str(req.session_hash))
|
|
|
377 |
|
378 |
gs_model = outputs["gaussian"][0]
|
379 |
mesh_model = outputs["mesh"][0]
|
380 |
+
with torch.no_grad():
|
381 |
+
color_images = render_video(gs_model, num_frames=1)["color"]
|
382 |
+
normal_images = render_video(mesh_model, num_frames=1)["normal"]
|
383 |
output_root = TMP_DIR
|
384 |
if req is not None:
|
385 |
output_root = os.path.join(output_root, str(req.session_hash))
|