| { |
| "module": "keras_hub.src.models.swin_transformer.swin_transformer_image_classifier", |
| "class_name": "SwinTransformerImageClassifier", |
| "config": { |
| "backbone": { |
| "module": "keras_hub.src.models.swin_transformer.swin_transformer_backbone", |
| "class_name": "SwinTransformerBackbone", |
| "config": { |
| "name": "swin_transformer_backbone", |
| "trainable": true, |
| "dtype": { |
| "module": "keras", |
| "class_name": "DTypePolicy", |
| "config": { |
| "name": "float32" |
| }, |
| "registered_name": null |
| }, |
| "image_shape": [ |
| 224, |
| 224, |
| 3 |
| ], |
| "patch_size": 4, |
| "embed_dim": 192, |
| "depths": [ |
| 2, |
| 2, |
| 18, |
| 2 |
| ], |
| "num_heads": [ |
| 6, |
| 12, |
| 24, |
| 48 |
| ], |
| "window_size": 7, |
| "mlp_ratio": 4.0, |
| "qkv_bias": true, |
| "dropout_rate": 0.0, |
| "attention_dropout": 0.0, |
| "drop_path": 0.1, |
| "patch_norm": true, |
| "data_format": "channels_last" |
| }, |
| "registered_name": "keras_hub>SwinTransformerBackbone" |
| }, |
| "preprocessor": { |
| "module": "keras_hub.src.models.swin_transformer.swin_transformer_image_classifier_preprocessor", |
| "class_name": "SwinTransformerImageClassifierPreprocessor", |
| "config": { |
| "name": "swin_transformer_image_classifier_preprocessor", |
| "trainable": true, |
| "dtype": { |
| "module": "keras", |
| "class_name": "DTypePolicy", |
| "config": { |
| "name": "float32" |
| }, |
| "registered_name": null |
| }, |
| "image_converter": { |
| "module": "keras_hub.src.models.swin_transformer.swin_transformer_image_converter", |
| "class_name": "SwinTransformerImageConverter", |
| "config": { |
| "name": "swin_transformer_image_converter", |
| "trainable": true, |
| "dtype": { |
| "module": "keras", |
| "class_name": "DTypePolicy", |
| "config": { |
| "name": "float32" |
| }, |
| "registered_name": null |
| }, |
| "image_size": [ |
| 224, |
| 224 |
| ], |
| "scale": [ |
| 0.017124753831663668, |
| 0.01750700280112045, |
| 0.017429193899782133 |
| ], |
| "offset": [ |
| -2.1179039301310043, |
| -2.0357142857142856, |
| -1.8044444444444445 |
| ], |
| "interpolation": "bicubic", |
| "antialias": true, |
| "crop_to_aspect_ratio": false, |
| "pad_to_aspect_ratio": false, |
| "bounding_box_format": "yxyx" |
| }, |
| "registered_name": "keras_hub>SwinTransformerImageConverter" |
| }, |
| "config_file": "preprocessor.json" |
| }, |
| "registered_name": "keras_hub>SwinTransformerImageClassifierPreprocessor" |
| }, |
| "name": "swin_transformer_image_classifier", |
| "num_classes": 1000, |
| "pooling": "gap", |
| "activation": null, |
| "dropout": 0.0 |
| }, |
| "registered_name": "keras_hub>SwinTransformerImageClassifier" |
| } |
