hujiecpp commited on
Commit
8a50897
·
1 Parent(s): 5505892

init project

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -584,10 +584,10 @@ def get_reconstructed_scene(outdir, filelist, schedule='linear', niter=300, min_
584
  fix_imgs.append(img)
585
  cogs = []
586
  for cog in scene.cogs:
587
- cog_cpu = cog.cpu()
588
  cogs.append(cog_cpu)
589
- focals = scene.get_focals().cpu()
590
- cams2world = scene.get_im_poses().cpu()
591
  pts3d = to_numpy(scene.get_pts3d())
592
  min_conf_thr = float(scene.conf_trf(torch.tensor(3.0)))
593
  msk = to_numpy(scene.get_masks())
 
584
  fix_imgs.append(img)
585
  cogs = []
586
  for cog in scene.cogs:
587
+ cog_cpu = cog.detach().cpu()
588
  cogs.append(cog_cpu)
589
+ focals = scene.get_focals().detach().cpu()
590
+ cams2world = scene.get_im_poses().detach().cpu()
591
  pts3d = to_numpy(scene.get_pts3d())
592
  min_conf_thr = float(scene.conf_trf(torch.tensor(3.0)))
593
  msk = to_numpy(scene.get_masks())