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
|