mattdangerw
commited on
Commit
•
e5ac0e3
1
Parent(s):
378608e
Upload folder using huggingface_hub
Browse files- README.md +20 -0
- config.json +24 -0
- image_converter.json +35 -0
- metadata.json +9 -0
- model.weights.h5 +3 -0
- preprocessor.json +53 -0
README.md
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
library_name: keras-hub
|
3 |
+
---
|
4 |
+
This is a [`ViT` model](https://keras.io/api/keras_hub/models/vi_t) uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends.
|
5 |
+
Model config:
|
6 |
+
* **name:** vi_t_backbone
|
7 |
+
* **trainable:** True
|
8 |
+
* **image_shape:** [224, 224, 3]
|
9 |
+
* **patch_size:** 32
|
10 |
+
* **num_layers:** 12
|
11 |
+
* **num_heads:** 12
|
12 |
+
* **hidden_dim:** 768
|
13 |
+
* **mlp_dim:** 3072
|
14 |
+
* **dropout_rate:** 0.0
|
15 |
+
* **attention_dropout:** 0.0
|
16 |
+
* **layer_norm_epsilon:** 1e-06
|
17 |
+
* **use_mha_bias:** True
|
18 |
+
* **use_mlp_bias:** True
|
19 |
+
|
20 |
+
This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.
|
config.json
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"module": "keras_hub.src.models.vit.vit_backbone",
|
3 |
+
"class_name": "ViTBackbone",
|
4 |
+
"config": {
|
5 |
+
"name": "vi_t_backbone",
|
6 |
+
"trainable": true,
|
7 |
+
"image_shape": [
|
8 |
+
224,
|
9 |
+
224,
|
10 |
+
3
|
11 |
+
],
|
12 |
+
"patch_size": 32,
|
13 |
+
"num_layers": 12,
|
14 |
+
"num_heads": 12,
|
15 |
+
"hidden_dim": 768,
|
16 |
+
"mlp_dim": 3072,
|
17 |
+
"dropout_rate": 0.0,
|
18 |
+
"attention_dropout": 0.0,
|
19 |
+
"layer_norm_epsilon": 1e-06,
|
20 |
+
"use_mha_bias": true,
|
21 |
+
"use_mlp_bias": true
|
22 |
+
},
|
23 |
+
"registered_name": "keras_hub>ViTBackbone"
|
24 |
+
}
|
image_converter.json
ADDED
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"module": "keras_hub.src.models.vit.vit_image_converter",
|
3 |
+
"class_name": "ViTImageConverter",
|
4 |
+
"config": {
|
5 |
+
"name": "vi_t_image_converter",
|
6 |
+
"trainable": true,
|
7 |
+
"dtype": {
|
8 |
+
"module": "keras",
|
9 |
+
"class_name": "DTypePolicy",
|
10 |
+
"config": {
|
11 |
+
"name": "float32"
|
12 |
+
},
|
13 |
+
"registered_name": null
|
14 |
+
},
|
15 |
+
"image_size": [
|
16 |
+
224,
|
17 |
+
224
|
18 |
+
],
|
19 |
+
"scale": 0.00392156862745098,
|
20 |
+
"offset": null,
|
21 |
+
"interpolation": "bilinear",
|
22 |
+
"crop_to_aspect_ratio": true,
|
23 |
+
"norm_mean": [
|
24 |
+
0.5,
|
25 |
+
0.5,
|
26 |
+
0.5
|
27 |
+
],
|
28 |
+
"norm_std": [
|
29 |
+
0.5,
|
30 |
+
0.5,
|
31 |
+
0.5
|
32 |
+
]
|
33 |
+
},
|
34 |
+
"registered_name": "keras_hub>ViTImageConverter"
|
35 |
+
}
|
metadata.json
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"keras_version": "3.7.0",
|
3 |
+
"keras_hub_version": "0.19.0",
|
4 |
+
"parameter_count": 87455232,
|
5 |
+
"date_saved": "2024-12-17@15:52:39",
|
6 |
+
"tasks": [
|
7 |
+
"ImageClassifier"
|
8 |
+
]
|
9 |
+
}
|
model.weights.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2982b47c019a5f5d53ae2af7be794cbbbeae3b01f901cdf4a54e2a5f94df6bf2
|
3 |
+
size 350266288
|
preprocessor.json
ADDED
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"module": "keras_hub.src.models.vit.vit_image_classifier_preprocessor",
|
3 |
+
"class_name": "ViTImageClassifierPreprocessor",
|
4 |
+
"config": {
|
5 |
+
"name": "vi_t_image_classifier_preprocessor",
|
6 |
+
"trainable": true,
|
7 |
+
"dtype": {
|
8 |
+
"module": "keras",
|
9 |
+
"class_name": "DTypePolicy",
|
10 |
+
"config": {
|
11 |
+
"name": "float32"
|
12 |
+
},
|
13 |
+
"registered_name": null
|
14 |
+
},
|
15 |
+
"image_converter": {
|
16 |
+
"module": "keras_hub.src.models.vit.vit_image_converter",
|
17 |
+
"class_name": "ViTImageConverter",
|
18 |
+
"config": {
|
19 |
+
"name": "vi_t_image_converter",
|
20 |
+
"trainable": true,
|
21 |
+
"dtype": {
|
22 |
+
"module": "keras",
|
23 |
+
"class_name": "DTypePolicy",
|
24 |
+
"config": {
|
25 |
+
"name": "float32"
|
26 |
+
},
|
27 |
+
"registered_name": null
|
28 |
+
},
|
29 |
+
"image_size": [
|
30 |
+
224,
|
31 |
+
224
|
32 |
+
],
|
33 |
+
"scale": 0.00392156862745098,
|
34 |
+
"offset": null,
|
35 |
+
"interpolation": "bilinear",
|
36 |
+
"crop_to_aspect_ratio": true,
|
37 |
+
"norm_mean": [
|
38 |
+
0.5,
|
39 |
+
0.5,
|
40 |
+
0.5
|
41 |
+
],
|
42 |
+
"norm_std": [
|
43 |
+
0.5,
|
44 |
+
0.5,
|
45 |
+
0.5
|
46 |
+
]
|
47 |
+
},
|
48 |
+
"registered_name": "keras_hub>ViTImageConverter"
|
49 |
+
},
|
50 |
+
"config_file": "preprocessor.json"
|
51 |
+
},
|
52 |
+
"registered_name": "keras_hub>ViTImageClassifierPreprocessor"
|
53 |
+
}
|