| { | |
| "library_name": "zeromodels", | |
| "zeromodels_version": "1.2.6", | |
| "model_module": "zeromodels.models.efficientdet", | |
| "model_class": "EfficientDetModel", | |
| "variant": "efficientdet_d2", | |
| "weights": "model.weights.h5", | |
| "schema_version": 2, | |
| "weight_dtype": "float32", | |
| "model_type": "efficientdet", | |
| "vision_config": { | |
| "backbone_name": "efficientnet_b2", | |
| "image_size": 768, | |
| "num_classes": 90, | |
| "min_level": 3, | |
| "max_level": 7, | |
| "num_scales": 3, | |
| "aspect_ratios": [ | |
| 1.0, | |
| 2.0, | |
| 0.5 | |
| ], | |
| "anchor_scale": 4.0, | |
| "fpn_num_filters": 112, | |
| "fpn_cell_repeats": 5, | |
| "box_class_repeats": 3, | |
| "act_type": "swish", | |
| "separable_conv": true, | |
| "apply_bn_for_resampling": true, | |
| "conv_after_downsample": false, | |
| "conv_bn_act_pattern": false, | |
| "fpn_weight_method": "fastattn", | |
| "survival_prob": null | |
| } | |
| } |