| { | |
| "module": "keras_hub.src.models.sam.sam_backbone", | |
| "class_name": "SAMBackbone", | |
| "config": { | |
| "name": "sam_backbone", | |
| "trainable": true, | |
| "image_encoder": { | |
| "module": "keras_hub.src.models.vit_det.vit_det_backbone", | |
| "class_name": "ViTDetBackbone", | |
| "config": { | |
| "name": "vi_t_det_backbone", | |
| "trainable": true, | |
| "image_shape": [ | |
| 1024, | |
| 1024, | |
| 3 | |
| ], | |
| "patch_size": 16, | |
| "hidden_size": 768, | |
| "num_layers": 12, | |
| "intermediate_dim": 3072, | |
| "num_heads": 12, | |
| "num_output_channels": 256, | |
| "use_bias": true, | |
| "use_abs_pos": true, | |
| "use_rel_pos": true, | |
| "window_size": 14, | |
| "global_attention_layer_indices": [ | |
| 2, | |
| 5, | |
| 8, | |
| 11 | |
| ], | |
| "layer_norm_epsilon": 1e-06 | |
| }, | |
| "registered_name": "keras_hub>ViTDetBackbone" | |
| }, | |
| "prompt_encoder": { | |
| "module": "keras_hub.src.models.sam.sam_prompt_encoder", | |
| "class_name": "SAMPromptEncoder", | |
| "config": { | |
| "name": "sam_prompt_encoder", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "hidden_size": 256, | |
| "image_embedding_size": [ | |
| 64, | |
| 64 | |
| ], | |
| "input_image_size": [ | |
| 1024, | |
| 1024 | |
| ], | |
| "mask_in_channels": 16, | |
| "activation": "gelu" | |
| }, | |
| "registered_name": "keras_hub>SAMPromptEncoder" | |
| }, | |
| "mask_decoder": { | |
| "module": "keras_hub.src.models.sam.sam_mask_decoder", | |
| "class_name": "SAMMaskDecoder", | |
| "config": { | |
| "name": "sam_mask_decoder", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "hidden_size": 256, | |
| "num_layers": 2, | |
| "intermediate_dim": 2048, | |
| "num_heads": 8, | |
| "embedding_dim": 256, | |
| "num_multimask_outputs": 3, | |
| "iou_head_depth": 3, | |
| "iou_head_hidden_dim": 256, | |
| "activation": "gelu" | |
| }, | |
| "registered_name": "keras_hub>SAMMaskDecoder" | |
| } | |
| }, | |
| "registered_name": "keras_hub>SAMBackbone" | |
| } |