RRoundTable commited on
Commit
e5a474c
·
1 Parent(s): 7f13a45

Add examples

Browse files
.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

  • SHA256: 100a0eb74a1ebfdfb4578afc8202b46f393b4aac307953646e230d4a3e77a213
  • Pointer size: 130 Bytes
  • Size of remote file: 20.1 kB