init project
Browse files
app.py
CHANGED
@@ -304,8 +304,8 @@ def get_cog_feats(images):
|
|
304 |
|
305 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
306 |
pe3r.sam2.to(device)
|
307 |
-
pe3r.siglip_processor.to(device)
|
308 |
-
pe3r.siglip.to(device)
|
309 |
|
310 |
cog_seg_maps = []
|
311 |
rev_cog_seg_maps = []
|
@@ -518,7 +518,7 @@ def get_3D_object_from_scene(outdir, text, threshold, scene, min_conf_thr, as_po
|
|
518 |
mask_sky, clean_depth, transparent_cams, cam_size):
|
519 |
|
520 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
521 |
-
pe3r.siglip_tokenizer.to(device)
|
522 |
|
523 |
|
524 |
texts = [text]
|
|
|
304 |
|
305 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
306 |
pe3r.sam2.to(device)
|
307 |
+
# pe3r.siglip_processor.to(device)
|
308 |
+
# pe3r.siglip.to(device)
|
309 |
|
310 |
cog_seg_maps = []
|
311 |
rev_cog_seg_maps = []
|
|
|
518 |
mask_sky, clean_depth, transparent_cams, cam_size):
|
519 |
|
520 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
521 |
+
# pe3r.siglip_tokenizer.to(device)
|
522 |
|
523 |
|
524 |
texts = [text]
|