Update app.py
Browse files
app.py
CHANGED
|
@@ -111,7 +111,7 @@ def get_reconstructed_scene(outdir, model, device):
|
|
| 111 |
model.decoder,
|
| 112 |
)
|
| 113 |
plyfile = os.path.join(outdir, "gaussians.ply")
|
| 114 |
-
splatfile = os.path.join(outdir, "gaussians.splat")
|
| 115 |
|
| 116 |
export_ply(
|
| 117 |
gaussians.means[0],
|
|
@@ -122,12 +122,12 @@ def get_reconstructed_scene(outdir, model, device):
|
|
| 122 |
Path(plyfile),
|
| 123 |
save_sh_dc_only=True,
|
| 124 |
)
|
| 125 |
-
splat_data = process_ply_to_splat(plyfile)
|
| 126 |
-
save_splat_file(splat_data, splatfile)
|
| 127 |
|
| 128 |
# Clean up
|
| 129 |
torch.cuda.empty_cache()
|
| 130 |
-
return
|
| 131 |
|
| 132 |
def extract_images(input_images, session_id):
|
| 133 |
|
|
|
|
| 111 |
model.decoder,
|
| 112 |
)
|
| 113 |
plyfile = os.path.join(outdir, "gaussians.ply")
|
| 114 |
+
# splatfile = os.path.join(outdir, "gaussians.splat")
|
| 115 |
|
| 116 |
export_ply(
|
| 117 |
gaussians.means[0],
|
|
|
|
| 122 |
Path(plyfile),
|
| 123 |
save_sh_dc_only=True,
|
| 124 |
)
|
| 125 |
+
# splat_data = process_ply_to_splat(plyfile)
|
| 126 |
+
# save_splat_file(splat_data, splatfile)
|
| 127 |
|
| 128 |
# Clean up
|
| 129 |
torch.cuda.empty_cache()
|
| 130 |
+
return plyfile, video, depth_colored
|
| 131 |
|
| 132 |
def extract_images(input_images, session_id):
|
| 133 |
|