init project
Browse files- modules/pe3r/demo.py +3 -4
modules/pe3r/demo.py
CHANGED
@@ -424,7 +424,7 @@ def get_cog_feats(images, pe3r, device):
|
|
424 |
|
425 |
return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
|
426 |
|
427 |
-
|
428 |
def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, niter, min_conf_thr,
|
429 |
as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,
|
430 |
scenegraph_type, winsize, refid):
|
@@ -511,7 +511,7 @@ def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, ni
|
|
511 |
|
512 |
return scene, outfile, imgs
|
513 |
|
514 |
-
|
515 |
def get_3D_object_from_scene(outdir, pe3r, silent, text, threshold, scene, min_conf_thr, as_pointcloud,
|
516 |
mask_sky, clean_depth, transparent_cams, cam_size):
|
517 |
|
@@ -549,10 +549,9 @@ def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
|
|
549 |
return winsize, refid
|
550 |
|
551 |
|
552 |
-
@spaces.GPU(duration=180)
|
553 |
def main_demo(tmpdirname, server_name, server_port, silent=False):
|
554 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
555 |
-
|
556 |
pe3r = Models(device)
|
557 |
# scene, outfile, imgs = get_reconstructed_scene(
|
558 |
# outdir=tmpdirname, pe3r=pe3r, device=device, silent=silent,
|
|
|
424 |
|
425 |
return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
|
426 |
|
427 |
+
@spaces.GPU(duration=180)
|
428 |
def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, niter, min_conf_thr,
|
429 |
as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,
|
430 |
scenegraph_type, winsize, refid):
|
|
|
511 |
|
512 |
return scene, outfile, imgs
|
513 |
|
514 |
+
@spaces.GPU(duration=180)
|
515 |
def get_3D_object_from_scene(outdir, pe3r, silent, text, threshold, scene, min_conf_thr, as_pointcloud,
|
516 |
mask_sky, clean_depth, transparent_cams, cam_size):
|
517 |
|
|
|
549 |
return winsize, refid
|
550 |
|
551 |
|
|
|
552 |
def main_demo(tmpdirname, server_name, server_port, silent=False):
|
553 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
554 |
+
|
555 |
pe3r = Models(device)
|
556 |
# scene, outfile, imgs = get_reconstructed_scene(
|
557 |
# outdir=tmpdirname, pe3r=pe3r, device=device, silent=silent,
|