generall93 jmzzomg commited on
Commit
0e77bf0
1 Parent(s): 92673de

unicom-vit-b-32 model card (#1)

Browse files

- unicom-vit-b-32 model card (9f48f41968f74ee944d8add3618e194d7b4e8176)


Co-authored-by: George <[email protected]>

Files changed (1) hide show
  1. README.md +29 -0
README.md ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: image-feature-extraction
4
+ ---
5
+
6
+ ONNX port of [Unicom](https://arxiv.org/abs/2304.05884) model from [open-metric-learning](https://github.com/OML-Team/open-metric-learning).
7
+
8
+ This model is intended to be used for similarity search.
9
+
10
+ ### Usage
11
+
12
+ Here's an example of performing inference using the model with [FastEmbed](https://github.com/qdrant/fastembed).
13
+
14
+ ```py
15
+ from fastembed import ImageEmbedding
16
+
17
+ images = [
18
+ "./path/to/image1.jpg",
19
+ "./path/to/image2.jpg",
20
+ ]
21
+
22
+ model = ImageEmbedding(model_name="Qdrant/Unicom-ViT-B-32")
23
+ embeddings = list(model.embed(images))
24
+
25
+ # [
26
+ # array([ 0.04177791, 0.0550059 , 0.00025418, 0.0252876 , ..., dtype=float32),
27
+ # array([2.23932182e-03, 4.68995124e-02, 3.28772422e-03, 7.57176951e-02, ...], dtype=float32)
28
+ # ]
29
+ ```