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())
|