{ "module": "keras_hub.src.models.d_fine.d_fine_object_detector", "class_name": "DFineObjectDetector", "config": { "backbone": { "module": "keras_hub.src.models.d_fine.d_fine_backbone", "class_name": "DFineBackbone", "config": { "name": "d_fine_backbone", "trainable": true, "backbone": { "module": "keras_hub.src.models.hgnetv2.hgnetv2_backbone", "class_name": "HGNetV2Backbone", "config": { "name": "hg_net_v2_backbone", "trainable": true, "depths": [ 3, 4, 6, 3 ], "embedding_size": 32, "hidden_sizes": [ 128, 256, 512, 1024 ], "stem_channels": [ 3, 16, 16 ], "hidden_act": "relu", "use_learnable_affine_block": true, "stackwise_stage_filters": [ [ 16, 16, 64, 1, 3, 3 ], [ 64, 32, 256, 1, 3, 3 ], [ 256, 64, 512, 2, 3, 5 ], [ 512, 128, 1024, 1, 3, 5 ] ], "apply_downsample": [ false, true, true, true ], "use_lightweight_conv_block": [ false, false, true, true ], "image_shape": [ null, null, 3 ], "out_features": [ "stage2", "stage3", "stage4" ], "data_format": "channels_last" }, "registered_name": "keras_hub>HGNetV2Backbone" }, "decoder_in_channels": [ 256, 256, 256 ], "encoder_hidden_dim": 256, "num_labels": 80, "num_denoising": 100, "learn_initial_query": false, "num_queries": 300, "anchor_image_size": [ 640, 640 ], "feat_strides": [ 8, 16, 32 ], "num_feature_levels": 3, "hidden_dim": 256, "encoder_in_channels": [ 256, 512, 1024 ], "encode_proj_layers": [ 2 ], "num_attention_heads": 8, "encoder_ffn_dim": 1024, "num_encoder_layers": 1, "hidden_expansion": 0.5, "depth_multiplier": 0.34, "eval_idx": -1, "box_noise_scale": 1.0, "label_noise_ratio": 0.5, "labels": null, "num_decoder_layers": 3, "decoder_attention_heads": 8, "decoder_ffn_dim": 1024, "decoder_method": "default", "decoder_n_points": [ 3, 6, 3 ], "lqe_hidden_dim": 64, "num_lqe_layers": 2, "seed": 0, "image_shape": [ null, null, 3 ], "data_format": "channels_last", "out_features": [ "stage2", "stage3", "stage4" ] }, "registered_name": "keras_hub>DFineBackbone" }, "preprocessor": { "module": "keras_hub.src.models.d_fine.d_fine_object_detector_preprocessor", "class_name": "DFineObjectDetectorPreprocessor", "config": { "name": "d_fine_object_detector_preprocessor", "trainable": true, "dtype": { "module": "keras", "class_name": "DTypePolicy", "config": { "name": "float32" }, "registered_name": null }, "image_converter": { "module": "keras_hub.src.models.d_fine.d_fine_image_converter", "class_name": "DFineImageConverter", "config": { "name": "d_fine_image_converter", "trainable": true, "dtype": { "module": "keras", "class_name": "DTypePolicy", "config": { "name": "float32" }, "registered_name": null }, "image_size": [ 640, 640 ], "scale": 0.00392156862745098, "offset": null, "interpolation": "bilinear", "antialias": false, "crop_to_aspect_ratio": true, "pad_to_aspect_ratio": false, "bounding_box_format": "yxyx" }, "registered_name": "keras_hub>DFineImageConverter" }, "config_file": "preprocessor.json" }, "registered_name": "keras_hub>DFineObjectDetectorPreprocessor" }, "name": "d_fine_object_detector", "num_classes": 80, "bounding_box_format": "yxyx", "matcher_class_cost": 2.0, "matcher_bbox_cost": 5.0, "matcher_ciou_cost": 2.0, "use_focal_loss": true, "matcher_alpha": 0.25, "matcher_gamma": 2.0, "weight_loss_vfl": 1.0, "weight_loss_bbox": 5.0, "weight_loss_ciou": 2.0, "weight_loss_fgl": 0.15, "weight_loss_ddf": 1.5, "ddf_temperature": 5.0, "prediction_decoder": { "module": "keras_hub.src.layers.modeling.non_max_supression", "class_name": "NonMaxSuppression", "config": { "name": "non_max_suppression", "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": 300 }, "registered_name": "keras_hub>NonMaxSuppression" } }, "registered_name": "keras_hub>DFineObjectDetector" }