Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
49b3e3d
1
Parent(s):
bc4bba8
Loading examples from huggingface hub
Browse files
demo.py
CHANGED
|
@@ -59,24 +59,28 @@ if __name__ == '__main__':
|
|
| 59 |
|
| 60 |
# Define example inputs and their corresponding precalculated outputs
|
| 61 |
examples = [
|
| 62 |
-
["
|
| 63 |
-
["
|
| 64 |
-
["
|
| 65 |
-
["
|
| 66 |
-
["
|
| 67 |
-
["
|
| 68 |
-
["
|
| 69 |
-
["
|
| 70 |
-
["
|
| 71 |
-
["
|
| 72 |
-
["
|
| 73 |
-
["
|
| 74 |
-
["
|
| 75 |
-
["
|
| 76 |
-
["
|
| 77 |
-
["
|
| 78 |
]
|
| 79 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 80 |
with tempfile.TemporaryDirectory(suffix='_mast3r_gradio_demo') as tmpdirname:
|
| 81 |
|
| 82 |
cache_path = os.path.join(tmpdirname, chkpt_tag)
|
|
|
|
| 59 |
|
| 60 |
# Define example inputs and their corresponding precalculated outputs
|
| 61 |
examples = [
|
| 62 |
+
["demo_examples/scannet++_1_img_1.jpg", "demo_examples/scannet++_1_img_2.jpg", "demo_examples/scannet++_1.ply"],
|
| 63 |
+
["demo_examples/scannet++_2_img_1.jpg", "demo_examples/scannet++_2_img_2.jpg", "demo_examples/scannet++_2.ply"],
|
| 64 |
+
["demo_examples/scannet++_3_img_1.jpg", "demo_examples/scannet++_3_img_2.jpg", "demo_examples/scannet++_3.ply"],
|
| 65 |
+
["demo_examples/scannet++_4_img_1.jpg", "demo_examples/scannet++_4_img_2.jpg", "demo_examples/scannet++_4.ply"],
|
| 66 |
+
["demo_examples/scannet++_5_img_1.jpg", "demo_examples/scannet++_5_img_2.jpg", "demo_examples/scannet++_5.ply"],
|
| 67 |
+
["demo_examples/scannet++_6_img_1.jpg", "demo_examples/scannet++_6_img_2.jpg", "demo_examples/scannet++_6.ply"],
|
| 68 |
+
["demo_examples/scannet++_7_img_1.jpg", "demo_examples/scannet++_7_img_2.jpg", "demo_examples/scannet++_7.ply"],
|
| 69 |
+
["demo_examples/scannet++_8_img_1.jpg", "demo_examples/scannet++_8_img_2.jpg", "demo_examples/scannet++_8.ply"],
|
| 70 |
+
["demo_examples/in_the_wild_1_img_1.jpg", "demo_examples/in_the_wild_1_img_2.jpg", "demo_examples/in_the_wild_1.ply"],
|
| 71 |
+
["demo_examples/in_the_wild_2_img_1.jpg", "demo_examples/in_the_wild_2_img_2.jpg", "demo_examples/in_the_wild_2.ply"],
|
| 72 |
+
["demo_examples/in_the_wild_3_img_1.jpg", "demo_examples/in_the_wild_3_img_2.jpg", "demo_examples/in_the_wild_3.ply"],
|
| 73 |
+
["demo_examples/in_the_wild_4_img_1.jpg", "demo_examples/in_the_wild_4_img_2.jpg", "demo_examples/in_the_wild_4.ply"],
|
| 74 |
+
["demo_examples/in_the_wild_5_img_1.jpg", "demo_examples/in_the_wild_5_img_2.jpg", "demo_examples/in_the_wild_5.ply"],
|
| 75 |
+
["demo_examples/in_the_wild_6_img_1.jpg", "demo_examples/in_the_wild_6_img_2.jpg", "demo_examples/in_the_wild_6.ply"],
|
| 76 |
+
["demo_examples/in_the_wild_7_img_1.jpg", "demo_examples/in_the_wild_7_img_2.jpg", "demo_examples/in_the_wild_7.ply"],
|
| 77 |
+
["demo_examples/in_the_wild_8_img_1.jpg", "demo_examples/in_the_wild_8_img_2.jpg", "demo_examples/in_the_wild_8.ply"],
|
| 78 |
]
|
| 79 |
|
| 80 |
+
for i in range(len(examples)):
|
| 81 |
+
for j in range(len(examples[i])):
|
| 82 |
+
examples[i][j] = hf_hub_download(repo_id=model_name, filename=examples[i][j])
|
| 83 |
+
|
| 84 |
with tempfile.TemporaryDirectory(suffix='_mast3r_gradio_demo') as tmpdirname:
|
| 85 |
|
| 86 |
cache_path = os.path.join(tmpdirname, chkpt_tag)
|