| { |
| "model": { |
| "model_type": "YOLO11x", |
| "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" |
| } |
| }, |
| "id2label": { |
| "0": "face" |
| }, |
| "label2id": { |
| "face": "0" |
| }, |
| "training": { |
| "epochs": 100, |
| "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" |
| } |
| } |
|
|
