mattdangerw
commited on
Commit
•
a4c8447
1
Parent(s):
545c7b5
Upload folder using huggingface_hub
Browse files- README.md +22 -0
- config.json +24 -0
- image_converter.json +35 -0
- metadata.json +9 -0
- model.weights.h5 +3 -0
- preprocessor.json +53 -0
- task.json +90 -0
- task.weights.h5 +3 -0
README.md
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
This model is related to a `ImageClassifier` task.
|
6 |
+
|
7 |
+
Model config:
|
8 |
+
* **name:** vi_t_backbone
|
9 |
+
* **trainable:** True
|
10 |
+
* **image_shape:** [384, 384, 3]
|
11 |
+
* **patch_size:** 32
|
12 |
+
* **num_layers:** 12
|
13 |
+
* **num_heads:** 12
|
14 |
+
* **hidden_dim:** 768
|
15 |
+
* **mlp_dim:** 3072
|
16 |
+
* **dropout_rate:** 0.0
|
17 |
+
* **attention_dropout:** 0.0
|
18 |
+
* **layer_norm_epsilon:** 1e-06
|
19 |
+
* **use_mha_bias:** True
|
20 |
+
* **use_mlp_bias:** True
|
21 |
+
|
22 |
+
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 |
+
384,
|
9 |
+
384,
|
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 |
+
384,
|
17 |
+
384
|
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": 87528192,
|
5 |
+
"date_saved": "2024-12-17@14:40:35",
|
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:42971bc1e792aaf25c138c32c5674449b1ebfa8f65fa16165b3e9ba7ee771347
|
3 |
+
size 350558128
|
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 |
+
384,
|
31 |
+
384
|
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 |
+
}
|
task.json
ADDED
@@ -0,0 +1,90 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"module": "keras_hub.src.models.vit.vit_image_classifier",
|
3 |
+
"class_name": "ViTImageClassifier",
|
4 |
+
"config": {
|
5 |
+
"backbone": {
|
6 |
+
"module": "keras_hub.src.models.vit.vit_backbone",
|
7 |
+
"class_name": "ViTBackbone",
|
8 |
+
"config": {
|
9 |
+
"name": "vi_t_backbone",
|
10 |
+
"trainable": true,
|
11 |
+
"image_shape": [
|
12 |
+
384,
|
13 |
+
384,
|
14 |
+
3
|
15 |
+
],
|
16 |
+
"patch_size": 32,
|
17 |
+
"num_layers": 12,
|
18 |
+
"num_heads": 12,
|
19 |
+
"hidden_dim": 768,
|
20 |
+
"mlp_dim": 3072,
|
21 |
+
"dropout_rate": 0.0,
|
22 |
+
"attention_dropout": 0.0,
|
23 |
+
"layer_norm_epsilon": 1e-06,
|
24 |
+
"use_mha_bias": true,
|
25 |
+
"use_mlp_bias": true
|
26 |
+
},
|
27 |
+
"registered_name": "keras_hub>ViTBackbone"
|
28 |
+
},
|
29 |
+
"preprocessor": {
|
30 |
+
"module": "keras_hub.src.models.vit.vit_image_classifier_preprocessor",
|
31 |
+
"class_name": "ViTImageClassifierPreprocessor",
|
32 |
+
"config": {
|
33 |
+
"name": "vi_t_image_classifier_preprocessor",
|
34 |
+
"trainable": true,
|
35 |
+
"dtype": {
|
36 |
+
"module": "keras",
|
37 |
+
"class_name": "DTypePolicy",
|
38 |
+
"config": {
|
39 |
+
"name": "float32"
|
40 |
+
},
|
41 |
+
"registered_name": null
|
42 |
+
},
|
43 |
+
"image_converter": {
|
44 |
+
"module": "keras_hub.src.models.vit.vit_image_converter",
|
45 |
+
"class_name": "ViTImageConverter",
|
46 |
+
"config": {
|
47 |
+
"name": "vi_t_image_converter",
|
48 |
+
"trainable": true,
|
49 |
+
"dtype": {
|
50 |
+
"module": "keras",
|
51 |
+
"class_name": "DTypePolicy",
|
52 |
+
"config": {
|
53 |
+
"name": "float32"
|
54 |
+
},
|
55 |
+
"registered_name": null
|
56 |
+
},
|
57 |
+
"image_size": [
|
58 |
+
384,
|
59 |
+
384
|
60 |
+
],
|
61 |
+
"scale": 0.00392156862745098,
|
62 |
+
"offset": null,
|
63 |
+
"interpolation": "bilinear",
|
64 |
+
"crop_to_aspect_ratio": true,
|
65 |
+
"norm_mean": [
|
66 |
+
0.5,
|
67 |
+
0.5,
|
68 |
+
0.5
|
69 |
+
],
|
70 |
+
"norm_std": [
|
71 |
+
0.5,
|
72 |
+
0.5,
|
73 |
+
0.5
|
74 |
+
]
|
75 |
+
},
|
76 |
+
"registered_name": "keras_hub>ViTImageConverter"
|
77 |
+
},
|
78 |
+
"config_file": "preprocessor.json"
|
79 |
+
},
|
80 |
+
"registered_name": "keras_hub>ViTImageClassifierPreprocessor"
|
81 |
+
},
|
82 |
+
"name": "vi_t_image_classifier",
|
83 |
+
"num_classes": 1000,
|
84 |
+
"pooling": "token",
|
85 |
+
"activation": null,
|
86 |
+
"dropout": 0.0,
|
87 |
+
"intermediate_dim": null
|
88 |
+
},
|
89 |
+
"registered_name": "keras_hub>ViTImageClassifier"
|
90 |
+
}
|
task.weights.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0b4ab8c9e3c8d5086c5c102c1350d4726bf04ce95b2de091490294279d8f143d
|
3 |
+
size 353650296
|