| { | |
| "backbone": "MobileNetV2", | |
| "input_shape": [ | |
| 224, | |
| 224, | |
| 3 | |
| ], | |
| "num_classes": 2, | |
| "anchor_boxes": [ | |
| [ | |
| 0.1, | |
| 0.1, | |
| 0.2, | |
| 0.2 | |
| ], | |
| [ | |
| 0.3, | |
| 0.3, | |
| 0.4, | |
| 0.4 | |
| ], | |
| [ | |
| 0.5, | |
| 0.5, | |
| 0.6, | |
| 0.6 | |
| ], | |
| [ | |
| 0.7, | |
| 0.7, | |
| 0.8, | |
| 0.8 | |
| ], | |
| [ | |
| 0.9, | |
| 0.9, | |
| 1.0, | |
| 1.0 | |
| ] | |
| ], | |
| "detection_head": { | |
| "num_filters": 256, | |
| "dropout_rate": 0.3 | |
| }, | |
| "optimizer": { | |
| "name": "Adam", | |
| "learning_rate": 0.001, | |
| "weight_decay": 0.0001 | |
| }, | |
| "loss": { | |
| "classification_loss": "binary_crossentropy", | |
| "localization_loss": "smooth_l1", | |
| "loss_weights": { | |
| "classification": 1.0, | |
| "localization": 1.0 | |
| } | |
| }, | |
| "training": { | |
| "epochs": 50, | |
| "early_stopping_patience": 10, | |
| "reduce_lr_patience": 5, | |
| "reduce_lr_factor": 0.1 | |
| } | |
| } |