| { | |
| "module": "keras_hub.src.models.retinanet.retinanet_object_detector", | |
| "class_name": "RetinaNetObjectDetector", | |
| "config": { | |
| "backbone": { | |
| "module": "keras_hub.src.models.retinanet.retinanet_backbone", | |
| "class_name": "RetinaNetBackbone", | |
| "config": { | |
| "name": "retina_net_backbone_1", | |
| "trainable": true, | |
| "image_encoder": { | |
| "module": "keras_hub.src.models.resnet.resnet_backbone", | |
| "class_name": "ResNetBackbone", | |
| "config": { | |
| "name": "res_net_backbone", | |
| "trainable": true, | |
| "input_conv_filters": [ | |
| 64 | |
| ], | |
| "input_conv_kernel_sizes": [ | |
| 7 | |
| ], | |
| "stackwise_num_filters": [ | |
| 64, | |
| 128, | |
| 256, | |
| 512 | |
| ], | |
| "stackwise_num_blocks": [ | |
| 3, | |
| 4, | |
| 6, | |
| 3 | |
| ], | |
| "stackwise_num_strides": [ | |
| 1, | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "block_type": "bottleneck_block", | |
| "use_pre_activation": false, | |
| "image_shape": [ | |
| null, | |
| null, | |
| 3 | |
| ] | |
| }, | |
| "registered_name": "keras_hub>ResNetBackbone" | |
| }, | |
| "min_level": 3, | |
| "max_level": 7, | |
| "use_p5": false, | |
| "use_fpn_batch_norm": false, | |
| "image_shape": [ | |
| null, | |
| null, | |
| 3 | |
| ] | |
| }, | |
| "registered_name": "keras_hub>RetinaNetBackbone" | |
| }, | |
| "preprocessor": { | |
| "module": "keras_hub.src.models.retinanet.retinanet_object_detector_preprocessor", | |
| "class_name": "RetinaNetObjectDetectorPreprocessor", | |
| "config": { | |
| "name": "retina_net_object_detector_preprocessor", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "image_converter": { | |
| "module": "keras_hub.src.models.retinanet.retinanet_image_converter", | |
| "class_name": "RetinaNetImageConverter", | |
| "config": { | |
| "name": "retina_net_image_converter", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "image_size": null, | |
| "scale": 0.00392156862745098, | |
| "offset": null, | |
| "interpolation": "bilinear", | |
| "crop_to_aspect_ratio": true, | |
| "norm_mean": [ | |
| 0.485, | |
| 0.456, | |
| 0.406 | |
| ], | |
| "norm_std": [ | |
| 0.229, | |
| 0.224, | |
| 0.225 | |
| ] | |
| }, | |
| "registered_name": "keras_hub>RetinaNetImageConverter" | |
| }, | |
| "config_file": "preprocessor.json" | |
| }, | |
| "registered_name": "keras_hub>RetinaNetObjectDetectorPreprocessor" | |
| }, | |
| "name": "retina_net_object_detector_1", | |
| "num_classes": 91, | |
| "use_prediction_head_norm": false, | |
| "pre_logits_num_conv_layers": 4, | |
| "bounding_box_format": "yxyx", | |
| "anchor_generator": { | |
| "module": "keras_hub.src.models.retinanet.anchor_generator", | |
| "class_name": "AnchorGenerator", | |
| "config": { | |
| "name": "anchor_generator_1", | |
| "min_level": 3, | |
| "max_level": 7, | |
| "num_scales": 3, | |
| "aspect_ratios": [ | |
| 0.5, | |
| 1.0, | |
| 2.0 | |
| ], | |
| "anchor_size": 4, | |
| "bounding_box_format": "yxyx", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| } | |
| }, | |
| "registered_name": "keras_hub>AnchorGenerator" | |
| }, | |
| "label_encoder": { | |
| "module": "keras_hub.src.models.retinanet.retinanet_label_encoder", | |
| "class_name": "RetinaNetLabelEncoder", | |
| "config": { | |
| "name": "retina_net_label_encoder_1", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "anchor_generator": { | |
| "module": "keras_hub.src.models.retinanet.anchor_generator", | |
| "class_name": "AnchorGenerator", | |
| "config": { | |
| "name": "anchor_generator_1", | |
| "min_level": 3, | |
| "max_level": 7, | |
| "num_scales": 3, | |
| "aspect_ratios": [ | |
| 0.5, | |
| 1.0, | |
| 2.0 | |
| ], | |
| "anchor_size": 4, | |
| "bounding_box_format": "yxyx", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| } | |
| }, | |
| "registered_name": "keras_hub>AnchorGenerator" | |
| }, | |
| "bounding_box_format": "yxyx", | |
| "encoding_format": "center_xywh", | |
| "positive_threshold": 0.5, | |
| "box_variance": [ | |
| 1.0, | |
| 1.0, | |
| 1.0, | |
| 1.0 | |
| ], | |
| "negative_threshold": 0.4, | |
| "background_class": -1.0, | |
| "ignore_class": -2.0 | |
| }, | |
| "registered_name": "RetinaNetLabelEncoder" | |
| }, | |
| "prediction_decoder": { | |
| "module": "keras_hub.src.models.retinanet.non_max_supression", | |
| "class_name": "NonMaxSuppression", | |
| "config": { | |
| "name": "non_max_suppression_1", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "bounding_box_format": "yxyx", | |
| "from_logits": true, | |
| "iou_threshold": 0.5, | |
| "confidence_threshold": 0.5, | |
| "max_detections": 100 | |
| }, | |
| "registered_name": "NonMaxSuppression" | |
| } | |
| }, | |
| "registered_name": "keras_hub>RetinaNetObjectDetector" | |
| } |