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]
|