alexnasa commited on
Commit
13a9517
·
verified ·
1 Parent(s): d744855

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 splatfile, video, depth_colored
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