init project
Browse files
app.py
CHANGED
|
@@ -38,10 +38,9 @@ from modules.pe3r.models import Models
|
|
| 38 |
import torchvision.transforms as tvf
|
| 39 |
|
| 40 |
silent = False
|
| 41 |
-
device = 'cuda' if torch.cuda.is_available() else
|
| 42 |
pe3r = Models(device) #
|
| 43 |
|
| 44 |
-
|
| 45 |
def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
|
| 46 |
cam_color=None, as_pointcloud=False,
|
| 47 |
transparent_cams=False):
|
|
@@ -433,6 +432,7 @@ def get_cog_feats(images):
|
|
| 433 |
|
| 434 |
return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
|
| 435 |
|
|
|
|
| 436 |
@spaces.GPU(duration=120)
|
| 437 |
def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
|
| 438 |
as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,
|
|
|
|
| 38 |
import torchvision.transforms as tvf
|
| 39 |
|
| 40 |
silent = False
|
| 41 |
+
device = 'cpu' # 'cuda' if torch.cuda.is_available() else
|
| 42 |
pe3r = Models(device) #
|
| 43 |
|
|
|
|
| 44 |
def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
|
| 45 |
cam_color=None, as_pointcloud=False,
|
| 46 |
transparent_cams=False):
|
|
|
|
| 432 |
|
| 433 |
return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
|
| 434 |
|
| 435 |
+
|
| 436 |
@spaces.GPU(duration=120)
|
| 437 |
def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
|
| 438 |
as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,
|