tillmannohm commited on
Commit
75fe452
·
verified ·
1 Parent(s): 060e00d

Upload 3 files

Browse files
Files changed (3) hide show
  1. README.md +110 -0
  2. config.yaml +146 -0
  3. model.onnx +3 -0
README.md ADDED
@@ -0,0 +1,110 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ - alexnet
13
+ - imagenet
14
+ - compute:light
15
+ - modality:image
16
+ base_model: torchvision/alexnet-imagenet1k-v1
17
+ datasets:
18
+ - imagenet-1k
19
+ metrics:
20
+ - cosine-similarity
21
+ model-index:
22
+ - name: alexnet_imagenet1k_4096
23
+ results: []
24
+ ---
25
+
26
+ # alexnet_imagenet1k_4096
27
+
28
+ alexnet_imagenet1k_4096 is a light image embedding model for fast first-pass visual grouping in EIDORA. Choose it when you want a small, classic baseline that runs comfortably on ordinary laptops.
29
+
30
+ ## Best For
31
+
32
+ - Fast baseline visual grouping.
33
+ - Small or exploratory image projects on ordinary laptops.
34
+ - Regression testing the EIDORA ONNX package pipeline.
35
+
36
+ ## Not Ideal For
37
+
38
+ - Fine-grained visual similarity where modern self-supervised models perform better.
39
+ - Text, video, or audio inputs.
40
+ - Production-quality semantic image retrieval when a stronger model is acceptable.
41
+
42
+ ## Compute Tier
43
+
44
+ Light: small download, low memory, faster CPU runtime. Intended for laptop CPU use and large first-pass projects.
45
+
46
+ ## Inputs
47
+
48
+ - `image`: required image input from `media_source`.
49
+
50
+ ## Output
51
+
52
+ The primary output is `embedding`, a float32 tensor shaped `[batch, 4096]`. Embeddings are already normalized and are intended for cosine similarity.
53
+
54
+ ## Usage In EIDORA
55
+
56
+ EIDORA shows this package as a light image embedding model in the Model Zoo. Use it for discovery maps, grouping, retrieval, and related embedding workflows.
57
+
58
+ ## Usage Examples
59
+
60
+ Download the complete package from Hugging Face and keep the generated files together:
61
+
62
+ ```python
63
+ from huggingface_hub import snapshot_download
64
+
65
+ package_dir = snapshot_download("eidora/alexnet_imagenet1k_4096")
66
+ # EIDORA reads config.yaml, README.md, and model.onnx from this folder.
67
+ ```
68
+
69
+ ## Preprocessing
70
+
71
+ - `image`: 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]`.
72
+
73
+ ## Authorship And Citation
74
+
75
+ This ONNX package was produced by EIDORA from the original AlexNet ImageNet model. EIDORA converted the model to ONNX and is not the original model creator. Please cite ImageNet Classification with Deep Convolutional Neural Networks and the original model repository when using this converted model.
76
+
77
+ Original model: https://proceedings.neurips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html
78
+
79
+ Original paper: https://proceedings.neurips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html
80
+
81
+ Authors: Alex Krizhevsky, Ilya Sutskever, Geoffrey E. Hinton
82
+
83
+ ```bibtex
84
+ @inproceedings{krizhevsky2012imagenet,
85
+ title={ImageNet Classification with Deep Convolutional Neural Networks},
86
+ author={Krizhevsky, Alex and Sutskever, Ilya and Hinton, Geoffrey E.},
87
+ booktitle={Advances in Neural Information Processing Systems},
88
+ year={2012}
89
+ }
90
+ ```
91
+
92
+ ## Training Data And Provenance
93
+
94
+ Base model: `torchvision/alexnet-imagenet1k-v1`. Source repository: https://pytorch.org/vision/stable/models/generated/torchvision.models.alexnet.html. Known training data: ImageNet-1K supervised classification data. Package payload size: 228020495 bytes.
95
+
96
+ ## Evaluation And Validation
97
+
98
+ 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 `eidora-model.yaml`.
99
+
100
+ ## Limitations And Safety
101
+
102
+ AlexNet is an older supervised ImageNet model. It is useful as a lightweight baseline, but modern CLIP, DINOv2, SigLIP, or domain-specific models will usually produce stronger semantic similarity.
103
+
104
+ ## License And Attribution
105
+
106
+ This package uses license `bsd-3-clause`. Upstream license: BSD-3-Clause for TorchVision code; ImageNet-trained weights distributed by PyTorch under their documented model terms. Converted to ONNX for EIDORA from the TorchVision AlexNet ImageNet weights.
107
+
108
+ ## Version
109
+
110
+ Package version: 1.0.0. ONNX opset: 17. Exporter: eidora-onnx-exporter 0.1.0.
config.yaml ADDED
@@ -0,0 +1,146 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ schema_version: 1
2
+ id: EIDORA/AlexNet_IN1k
3
+ name: AlexNet_IN1k
4
+ version: 1.0.0
5
+ model_family: AlexNet
6
+ backend: onnx
7
+ runtime:
8
+ adapter: onnx_image
9
+ execution_provider: CPUExecutionProvider
10
+ model_path: model.onnx
11
+ input_names:
12
+ - pixel_values
13
+ output_name: embedding
14
+ onnxruntime:
15
+ opset: 17
16
+ tested_versions: '>=1.17,<2'
17
+ artifact:
18
+ path: model.onnx
19
+ sha256: 3e1277374f9da12a0c92084e19003dd541a792fccd5b4c0fd2bc4bfa54b28cc6
20
+ package_size_bytes: 228020495
21
+ inputs:
22
+ - id: image
23
+ modality: image
24
+ label: Images
25
+ required: true
26
+ source_kind: media_source
27
+ requirements:
28
+ color_space: RGB
29
+ layout: NCHW
30
+ width: 224
31
+ height: 224
32
+ preprocess:
33
+ image:
34
+ resize:
35
+ mode: resize_shorter_edge_then_center_crop
36
+ resize_size: 256
37
+ crop_width: 224
38
+ crop_height: 224
39
+ interpolation: bilinear
40
+ source: TorchVision AlexNet_Weights.IMAGENET1K_V1.transforms
41
+ rescale: 1/255
42
+ normalize:
43
+ mean:
44
+ - 0.485
45
+ - 0.456
46
+ - 0.406
47
+ std:
48
+ - 0.229
49
+ - 0.224
50
+ - 0.225
51
+ inside_onnx: true
52
+ embedding:
53
+ dimensions: 4096
54
+ feature_type: embedding
55
+ pooling: fc_head
56
+ normalized: true
57
+ similarity: cosine
58
+ output_name: embedding
59
+ dtype: float32
60
+ shape:
61
+ - batch
62
+ - 4096
63
+ display:
64
+ summary: 'Light: classic ImageNet visual embeddings for fast first-pass grouping
65
+ on laptops.'
66
+ compute_tier: light
67
+ modality_labels:
68
+ - image
69
+ recommended_batch_size: 8
70
+ validation:
71
+ fixtures:
72
+ - id: image_tensor_001
73
+ input_shape:
74
+ - 1
75
+ - 3
76
+ - 224
77
+ - 224
78
+ expected_shape:
79
+ - 1
80
+ - 4096
81
+ seed: 17
82
+ checks:
83
+ load_with: onnxruntime
84
+ execution_provider: CPUExecutionProvider
85
+ output_dtype: float32
86
+ finite: true
87
+ normalized_l2_range:
88
+ - 0.99
89
+ - 1.01
90
+ provenance:
91
+ base_model: torchvision/alexnet-imagenet1k-v1
92
+ source_repository: https://pytorch.org/vision/stable/models/generated/torchvision.models.alexnet.html
93
+ original_model_name: AlexNet ImageNet
94
+ original_model_url: https://proceedings.neurips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html
95
+ authors:
96
+ - Alex Krizhevsky
97
+ - Ilya Sutskever
98
+ - Geoffrey E. Hinton
99
+ paper_title: ImageNet Classification with Deep Convolutional Neural Networks
100
+ paper_url: https://proceedings.neurips.cc/paper/2012/hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html
101
+ upstream_license: BSD-3-Clause for TorchVision code; ImageNet-trained weights distributed
102
+ by PyTorch under their documented model terms.
103
+ training_data: ImageNet-1K supervised classification data.
104
+ citation: "@inproceedings{krizhevsky2012imagenet,\n title={ImageNet Classification\
105
+ \ with Deep Convolutional Neural Networks},\n author={Krizhevsky, Alex and Sutskever,\
106
+ \ Ilya and Hinton, Geoffrey E.},\n booktitle={Advances in Neural Information\
107
+ \ Processing Systems},\n year={2012}\n}\n"
108
+ conversion_note: EIDORA produced this ONNX conversion and is not the original model
109
+ creator.
110
+ export_date: '2026-07-10'
111
+ exporter_version: eidora-onnx-exporter 0.1.0
112
+ model_card:
113
+ best_for:
114
+ - Fast baseline visual grouping.
115
+ - Small or exploratory image projects on ordinary laptops.
116
+ - Regression testing the EIDORA ONNX package pipeline.
117
+ not_ideal_for:
118
+ - Fine-grained visual similarity where modern self-supervised models perform better.
119
+ - Text, video, or audio inputs.
120
+ - Production-quality semantic image retrieval when a stronger model is acceptable.
121
+ limitations: AlexNet is an older supervised ImageNet model. It is useful as a lightweight
122
+ baseline, but modern CLIP, DINOv2, SigLIP, or domain-specific models will usually
123
+ produce stronger semantic similarity.
124
+ license:
125
+ id: bsd-3-clause
126
+ attribution: Converted to ONNX for EIDORA from the TorchVision AlexNet ImageNet
127
+ weights.
128
+ huggingface:
129
+ org: eidora
130
+ repo_name: alexnet_imagenet1k_4096
131
+ pipeline_tag: feature-extraction
132
+ tags:
133
+ - eidora
134
+ - eidora-model-zoo
135
+ - onnx
136
+ - onnxruntime
137
+ - embeddings
138
+ - image
139
+ - alexnet
140
+ - imagenet
141
+ - compute:light
142
+ - modality:image
143
+ datasets:
144
+ - imagenet-1k
145
+ metrics:
146
+ - cosine-similarity
model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e1277374f9da12a0c92084e19003dd541a792fccd5b4c0fd2bc4bfa54b28cc6
3
+ size 228019986