deven96 jmzzomg commited on
Commit
fe89179
·
0 Parent(s):

Duplicate from Qdrant/resnet50-onnx

Browse files

Co-authored-by: George <jmzzomg@users.noreply.huggingface.co>

Files changed (5) hide show
  1. .gitattributes +35 -0
  2. README.md +31 -0
  3. config.json +1 -0
  4. model.onnx +3 -0
  5. preprocessor_config.json +43 -0
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: image-classification
4
+ ---
5
+
6
+ ONNX port of [microsoft/resnet-50](https://huggingface.co/microsoft/resnet-50).
7
+
8
+ This model is intended to be used for image classification and similarity searches.
9
+
10
+ You can find the ONNX port implementation [here](https://github.com/qdrant/fastembed/blob/main/experiments/Example.%20Convert%20Resnet50%20to%20ONNX.ipynb)
11
+
12
+ ### Usage
13
+
14
+ Here's an example of performing inference using the model with [FastEmbed](https://github.com/qdrant/fastembed).
15
+
16
+ ```py
17
+ from fastembed import ImageEmbedding
18
+
19
+ images = [
20
+ "./path/to/image1.jpg",
21
+ "./path/to/image2.jpg",
22
+ ]
23
+
24
+ model = ImageEmbedding(model_name="Qdrant/resnet50-onnx")
25
+ embeddings = list(model.embed(images))
26
+
27
+ # [
28
+ # array([-0.1115, 0.0097, 0.0052, 0.0195, ...], dtype=float32),
29
+ # array([-0.1019, 0.0635, -0.0332, 0.0522, ...], dtype=float32)
30
+ # ]
31
+ ```
config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {}
model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f0558b775c85b568012f7174cadda33fb4b4e5a253ce17e5d839f123e0bc7e6
3
+ size 93961728
preprocessor_config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_processor_class": null,
3
+ "do_resize": true,
4
+ "size": {
5
+ "shortest_edge": 224
6
+ },
7
+ "crop_pct": 0.875,
8
+ "resample": 3,
9
+ "do_rescale": true,
10
+ "rescale_factor": 0.00392156862745098,
11
+ "do_normalize": true,
12
+ "image_mean": [
13
+ 0.485,
14
+ 0.456,
15
+ 0.406
16
+ ],
17
+ "image_std": [
18
+ 0.229,
19
+ 0.224,
20
+ 0.225
21
+ ],
22
+ "_valid_processor_keys": [
23
+ "images",
24
+ "do_resize",
25
+ "size",
26
+ "crop_pct",
27
+ "resample",
28
+ "do_rescale",
29
+ "rescale_factor",
30
+ "do_normalize",
31
+ "image_mean",
32
+ "image_std",
33
+ "return_tensors",
34
+ "data_format",
35
+ "input_data_format"
36
+ ],
37
+ "image_processor_type": "ConvNextFeatureExtractor",
38
+ "do_center_crop": true,
39
+ "crop_size": {
40
+ "height": 224,
41
+ "width": 224
42
+ }
43
+ }