tillmannohm commited on
Commit
a0a6cee
·
verified ·
1 Parent(s): ad25c33

Upload 3 files

Browse files
Files changed (3) hide show
  1. README.md +96 -0
  2. config.yaml +151 -0
  3. model.onnx +3 -0
README.md ADDED
@@ -0,0 +1,96 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: onnx
3
+ pipeline_tag: feature-extraction
4
+ license: bsd-3-clause
5
+ tags:
6
+ - eidora
7
+ - eidora-model-zoo
8
+ - onnx
9
+ - onnxruntime
10
+ - embeddings
11
+ - image
12
+ - resnet
13
+ - compute:medium
14
+ - modality:image
15
+ base_model: torchvision/resnet50-imagenet1k-v2
16
+ datasets:
17
+ - imagenet-1k
18
+ metrics:
19
+ - cosine-similarity
20
+ model-index:
21
+ - name: RESNET50_2048
22
+ results: []
23
+ ---
24
+
25
+ # RESNET50_2048
26
+
27
+ RESNET50_2048 is a medium image embedding model for dependable visual grouping in EIDORA.
28
+
29
+ ## Best For
30
+
31
+ - General image grouping and retrieval.
32
+ - Projects that need a well-established visual baseline.
33
+
34
+ ## Not Ideal For
35
+
36
+ - Text, video, or audio inputs.
37
+ - Fine-grained semantic retrieval where self-supervised models are preferred.
38
+
39
+ ## Compute Tier
40
+
41
+ Medium: better quality or broader domain coverage with moderate runtime cost. Intended for recent laptops and desktops.
42
+
43
+ ## Inputs
44
+
45
+ - `image`: required image input from `media_source`.
46
+
47
+ ## Output
48
+
49
+ The primary output is `embedding`, a float32 tensor shaped `[batch, 2048]`. Embeddings are already normalized and are intended for cosine similarity.
50
+
51
+ ## Usage In EIDORA
52
+
53
+ EIDORA shows this package as a medium image embedding model in the Model Zoo. Use it for discovery maps, grouping, retrieval, and related embedding workflows.
54
+
55
+ ## Preprocessing
56
+
57
+ - `image`: resize mode defaults to `crop_center`; rescale uses `1/255`; normalize inside the ONNX graph with mean `[0.485, 0.456, 0.406]` and std `[0.229, 0.224, 0.225]`.
58
+
59
+ ## Authorship And Citation
60
+
61
+ This ONNX package was produced by EIDORA from the original ResNet-50 ImageNet model. EIDORA converted the model to ONNX and is not the original model creator. Please cite Deep Residual Learning for Image Recognition and the original model repository when using this converted model.
62
+
63
+ Original model: https://github.com/pytorch/vision
64
+
65
+ Original paper: https://arxiv.org/abs/1512.03385
66
+
67
+ Authors: Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun
68
+
69
+ ```bibtex
70
+ @inproceedings{he2016deep,
71
+ title={Deep Residual Learning for Image Recognition},
72
+ author={He, Kaiming and Zhang, Xiangyu and Ren, Shaoqing and Sun, Jian},
73
+ booktitle={CVPR},
74
+ year={2016}
75
+ }
76
+ ```
77
+
78
+ ## Training Data And Provenance
79
+
80
+ Base model: `torchvision/resnet50-imagenet1k-v2`. Source repository: https://pytorch.org/vision/stable/models/generated/torchvision.models.resnet50.html. Known training data: ImageNet-1K supervised classification data. Package payload size: 93951417 bytes.
81
+
82
+ ## Evaluation And Validation
83
+
84
+ The package validation checks that the ONNX graph loads with ONNX Runtime CPU execution, runs the declared fixtures, returns finite float32 embeddings with the declared shape, and matches the artifact hash recorded in `config.yaml`.
85
+
86
+ ## Limitations And Safety
87
+
88
+ ResNet-50 is trained for ImageNet classification, so its embeddings may underperform specialized or self-supervised models on some visual domains.
89
+
90
+ ## License And Attribution
91
+
92
+ This package uses license `bsd-3-clause`. Upstream license: BSD-3-Clause for TorchVision code; ImageNet weights are distributed by PyTorch under documented model terms. Converted to ONNX for EIDORA from TorchVision ResNet-50 ImageNet weights.
93
+
94
+ ## Version
95
+
96
+ Package version: 1.0.0. ONNX opset: 17. Exporter: eidora-onnx-exporter 0.1.0.
config.yaml ADDED
@@ -0,0 +1,151 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ schema_version: 1
2
+ id: EIDORA/ResNet50_IN1k
3
+ name: ResNet50_IN1k
4
+ version: 1.0.0
5
+ model_family: ResNet
6
+ backend: onnx
7
+ file_size: 94 MB
8
+ runtime:
9
+ adapter: onnx_image
10
+ execution_provider: CPUExecutionProvider
11
+ model_path: model.onnx
12
+ input_names:
13
+ - pixel_values
14
+ output_name: embedding
15
+ onnxruntime:
16
+ opset: 17
17
+ tested_versions: '>=1.17,<2'
18
+ artifact:
19
+ path: model.onnx
20
+ sha256: 09c59c1eba07408c5174b5fd58774bd16495cf0503f0406b0accfb2502ae6df0
21
+ package_size_bytes: 93951417
22
+ inputs:
23
+ - id: image
24
+ modality: image
25
+ label: Images
26
+ required: true
27
+ source_kind: media_source
28
+ requirements:
29
+ color_space: RGB
30
+ layout: NCHW
31
+ width: 224
32
+ height: 224
33
+ preprocess:
34
+ image:
35
+ resize_mode:
36
+ label: Image preprocessing
37
+ kind: choice
38
+ choices:
39
+ - distort
40
+ - crop_center
41
+ - add_padding
42
+ default: crop_center
43
+ visible: true
44
+ resize:
45
+ mode: resize_shorter_edge_then_center_crop
46
+ resize_size: 256
47
+ crop_width: 224
48
+ crop_height: 224
49
+ interpolation: bilinear
50
+ source: TorchVision ResNet50_Weights.IMAGENET1K_V2.transforms
51
+ rescale: 1/255
52
+ normalize:
53
+ mean:
54
+ - 0.485
55
+ - 0.456
56
+ - 0.406
57
+ std:
58
+ - 0.229
59
+ - 0.224
60
+ - 0.225
61
+ inside_onnx: true
62
+ embedding:
63
+ dimensions: 2048
64
+ feature_type: embedding
65
+ pooling: feature_layer
66
+ normalized: true
67
+ similarity: cosine
68
+ output_name: embedding
69
+ dtype: float32
70
+ shape:
71
+ - batch
72
+ - 2048
73
+ display:
74
+ summary: 'Medium: reliable ImageNet visual embeddings with moderate CPU runtime.'
75
+ compute_tier: medium
76
+ modality_labels:
77
+ - image
78
+ recommended_batch_size: 8
79
+ validation:
80
+ fixtures:
81
+ - id: image_tensor_001
82
+ input_shape:
83
+ - 1
84
+ - 3
85
+ - 224
86
+ - 224
87
+ expected_shape:
88
+ - 1
89
+ - 2048
90
+ seed: 51
91
+ checks:
92
+ load_with: onnxruntime
93
+ execution_provider: CPUExecutionProvider
94
+ output_dtype: float32
95
+ finite: true
96
+ normalized_l2_range:
97
+ - 0.99
98
+ - 1.01
99
+ provenance:
100
+ base_model: torchvision/resnet50-imagenet1k-v2
101
+ source_repository: https://pytorch.org/vision/stable/models/generated/torchvision.models.resnet50.html
102
+ original_model_name: ResNet-50 ImageNet
103
+ original_model_url: https://github.com/pytorch/vision
104
+ authors:
105
+ - Kaiming He
106
+ - Xiangyu Zhang
107
+ - Shaoqing Ren
108
+ - Jian Sun
109
+ paper_title: Deep Residual Learning for Image Recognition
110
+ paper_url: https://arxiv.org/abs/1512.03385
111
+ upstream_license: BSD-3-Clause for TorchVision code; ImageNet weights are distributed
112
+ by PyTorch under documented model terms.
113
+ training_data: ImageNet-1K supervised classification data.
114
+ citation: "@inproceedings{he2016deep,\n title={Deep Residual Learning for Image\
115
+ \ Recognition},\n author={He, Kaiming and Zhang, Xiangyu and Ren, Shaoqing and\
116
+ \ Sun, Jian},\n booktitle={CVPR},\n year={2016}\n}\n"
117
+ conversion_note: EIDORA produced this ONNX conversion and is not the original model
118
+ creator.
119
+ export_date: '2026-07-14'
120
+ exporter_version: eidora-onnx-exporter 0.1.0
121
+ model_card:
122
+ best_for:
123
+ - General image grouping and retrieval.
124
+ - Projects that need a well-established visual baseline.
125
+ not_ideal_for:
126
+ - Text, video, or audio inputs.
127
+ - Fine-grained semantic retrieval where self-supervised models are preferred.
128
+ limitations: ResNet-50 is trained for ImageNet classification, so its embeddings
129
+ may underperform specialized or self-supervised models on some visual domains.
130
+ license:
131
+ id: bsd-3-clause
132
+ attribution: Converted to ONNX for EIDORA from TorchVision ResNet-50 ImageNet
133
+ weights.
134
+ huggingface:
135
+ org: eidora
136
+ repo_name: RESNET50_2048
137
+ pipeline_tag: feature-extraction
138
+ tags:
139
+ - eidora
140
+ - eidora-model-zoo
141
+ - onnx
142
+ - onnxruntime
143
+ - embeddings
144
+ - image
145
+ - resnet
146
+ - compute:medium
147
+ - modality:image
148
+ datasets:
149
+ - imagenet-1k
150
+ metrics:
151
+ - cosine-similarity
model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09c59c1eba07408c5174b5fd58774bd16495cf0503f0406b0accfb2502ae6df0
3
+ size 93950918