init project
Browse files
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())
|