mattdangerw commited on
Commit
2841359
1 Parent(s): 14aad16

Upload folder using huggingface_hub

Browse files
Files changed (8) hide show
  1. README.md +22 -0
  2. config.json +24 -0
  3. image_converter.json +35 -0
  4. metadata.json +9 -0
  5. model.weights.h5 +3 -0
  6. preprocessor.json +53 -0
  7. task.json +90 -0
  8. 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:** 24
13
+ * **num_heads:** 16
14
+ * **hidden_dim:** 1024
15
+ * **mlp_dim:** 4096
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": 24,
14
+ "num_heads": 16,
15
+ "hidden_dim": 1024,
16
+ "mlp_dim": 4096,
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": 305607680,
5
+ "date_saved": "2024-12-17@15:46:36",
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:69492f5a30d118d434b6b7fd04a6108070bbafd8511b50cd417f63b67d65c225
3
+ size 1223296824
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": 24,
18
+ "num_heads": 16,
19
+ "hidden_dim": 1024,
20
+ "mlp_dim": 4096,
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:c65350772c0ea83325f277b5ba72c5a53dd8b7f1ce07003f42008fcdc4072899
3
+ size 1227412992