Spaces:
Runtime error
Runtime error
RRoundTable
commited on
Commit
·
e5a474c
1
Parent(s):
7f13a45
Add examples
Browse files- .gitattributes +6 -0
- app.py +13 -0
- examples.py +15 -0
- examples/embeddings/dogs.pickle +3 -0
- examples/image_zipfile/dogs.zip +3 -0
- examples/images/dogs-1.jpg +3 -0
.gitattributes
CHANGED
@@ -32,3 +32,9 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
32 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
33 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
33 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
34 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
35 |
+
examples/images/dogs-1.jpg filter=lfs diff=lfs merge=lfs -text
|
36 |
+
examples/embeddings filter=lfs diff=lfs merge=lfs -text
|
37 |
+
examples/embeddings/dogs.pickle filter=lfs diff=lfs merge=lfs -text
|
38 |
+
examples/image_zipfile filter=lfs diff=lfs merge=lfs -text
|
39 |
+
examples/image_zipfile/dogs.zip filter=lfs diff=lfs merge=lfs -text
|
40 |
+
examples/images filter=lfs diff=lfs merge=lfs -text
|
app.py
CHANGED
@@ -10,6 +10,7 @@ import cv2
|
|
10 |
import pickle
|
11 |
import zipfile
|
12 |
import faiss
|
|
|
13 |
|
14 |
|
15 |
DINOV2_REPO = "facebookresearch/dinov2"
|
@@ -140,6 +141,18 @@ with gr.Blocks() as demo:
|
|
140 |
outputs=output_images + distances,
|
141 |
)
|
142 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
143 |
with gr.Tab("Image Embedding with database"):
|
144 |
with gr.Row():
|
145 |
embedding_btn = gr.Button(value="Image Embedding")
|
|
|
10 |
import pickle
|
11 |
import zipfile
|
12 |
import faiss
|
13 |
+
from examples import examples
|
14 |
|
15 |
|
16 |
DINOV2_REPO = "facebookresearch/dinov2"
|
|
|
141 |
outputs=output_images + distances,
|
142 |
)
|
143 |
|
144 |
+
gr.Examples(
|
145 |
+
examples=examples,
|
146 |
+
inputs=[
|
147 |
+
image_embedding_file,
|
148 |
+
image_zip_file,
|
149 |
+
input_image,
|
150 |
+
],
|
151 |
+
outputs=output_images + distances,
|
152 |
+
fn=instance_recognition,
|
153 |
+
run_on_click=True,
|
154 |
+
)
|
155 |
+
|
156 |
with gr.Tab("Image Embedding with database"):
|
157 |
with gr.Row():
|
158 |
embedding_btn = gr.Button(value="Image Embedding")
|
examples.py
ADDED
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import os
|
2 |
+
|
3 |
+
examples = [
|
4 |
+
[
|
5 |
+
os.path.join(
|
6 |
+
os.path.dirname(__file__),
|
7 |
+
"examples/embeddings/dogs.pickle"
|
8 |
+
),
|
9 |
+
os.path.join(
|
10 |
+
os.path.dirname(__file__),
|
11 |
+
"examples/image_zipfile/dogs.zip"
|
12 |
+
),
|
13 |
+
os.path.join(os.path.dirname(__file__), "examples/images/dogs-1.jpg"),
|
14 |
+
],
|
15 |
+
]
|
examples/embeddings/dogs.pickle
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:90efb63db02ee3f404a7502f57baf666d4dbef3fe54f680211045317834e08ed
|
3 |
+
size 417128
|
examples/image_zipfile/dogs.zip
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:51ab18a7197eb1f3f54abf55e320bed29fdc703d023bdea7ce4a40eeb26df4f4
|
3 |
+
size 770185
|
examples/images/dogs-1.jpg
ADDED
Git LFS Details
|