| { | |
| "model": { | |
| "type": "YOLO11n", | |
| "pretrained": true, | |
| "num_classes": 1, | |
| "input_size": 640 | |
| }, | |
| "dataset": { | |
| "name": "WIDER FACE", | |
| "train": { | |
| "images": "WIDER_train/images", | |
| "annotations": "WIDER_train/annotations" | |
| }, | |
| "val": { | |
| "images": "WIDER_val/images", | |
| "annotations": "WIDER_val/annotations" | |
| } | |
| }, | |
| "training": { | |
| "epochs": 225, | |
| "batch_size": 16, | |
| "learning_rate": 0.001, | |
| "optimizer": "Adam", | |
| "momentum": 0.9, | |
| "weight_decay": 0.0005, | |
| "scheduler": { | |
| "type": "StepLR", | |
| "step_size": 30, | |
| "gamma": 0.1 | |
| } | |
| }, | |
| "augmentation": { | |
| "horizontal_flip": true, | |
| "vertical_flip": false, | |
| "rotation": 10, | |
| "scale": [0.8, 1.2], | |
| "shear": 2 | |
| }, | |
| "evaluation": { | |
| "interval": 1, | |
| "metrics": ["mAP", "precision", "recall"] | |
| }, | |
| "save": { | |
| "checkpoint_dir": "checkpoints/", | |
| "best_model": "model.pth" | |
| } | |
| } | |