| { | |
| "module": "keras_hub.src.models.vit.vit_image_classifier", | |
| "class_name": "ViTImageClassifier", | |
| "config": { | |
| "backbone": { | |
| "module": "keras_hub.src.models.vit.vit_backbone", | |
| "class_name": "ViTBackbone", | |
| "config": { | |
| "name": "vi_t_backbone", | |
| "trainable": true, | |
| "image_shape": [ | |
| 384, | |
| 384, | |
| 3 | |
| ], | |
| "patch_size": [ | |
| 16, | |
| 16 | |
| ], | |
| "num_layers": 12, | |
| "num_heads": 12, | |
| "hidden_dim": 768, | |
| "mlp_dim": 3072, | |
| "dropout_rate": 0.0, | |
| "attention_dropout": 0.0, | |
| "layer_norm_epsilon": 1e-06, | |
| "use_mha_bias": true, | |
| "use_mlp_bias": true, | |
| "use_class_token": true, | |
| "use_patch_bias": true | |
| }, | |
| "registered_name": "keras_hub>ViTBackbone" | |
| }, | |
| "preprocessor": { | |
| "module": "keras_hub.src.models.vit.vit_image_classifier_preprocessor", | |
| "class_name": "ViTImageClassifierPreprocessor", | |
| "config": { | |
| "name": "vi_t_image_classifier_preprocessor", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "image_converter": { | |
| "module": "keras_hub.src.models.vit.vit_image_converter", | |
| "class_name": "ViTImageConverter", | |
| "config": { | |
| "name": "vi_t_image_converter", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "image_size": [ | |
| 384, | |
| 384 | |
| ], | |
| "scale": [ | |
| 0.00784313725490196, | |
| 0.00784313725490196, | |
| 0.00784313725490196 | |
| ], | |
| "offset": [ | |
| -1.0, | |
| -1.0, | |
| -1.0 | |
| ], | |
| "interpolation": "bilinear", | |
| "antialias": false, | |
| "crop_to_aspect_ratio": true, | |
| "pad_to_aspect_ratio": false, | |
| "bounding_box_format": "yxyx" | |
| }, | |
| "registered_name": "keras_hub>ViTImageConverter" | |
| }, | |
| "config_file": "preprocessor.json" | |
| }, | |
| "registered_name": "keras_hub>ViTImageClassifierPreprocessor" | |
| }, | |
| "name": "vi_t_image_classifier", | |
| "num_classes": 1000, | |
| "pooling": "token", | |
| "activation": null, | |
| "dropout": 0.0, | |
| "intermediate_dim": null | |
| }, | |
| "registered_name": "keras_hub>ViTImageClassifier" | |
| } |