Hugging Face's logo

AdamCodd
/
YOLOv11n-face-detection

Object Detection
ONNX
PyTorch
Model card Files
xet
AdamCodd commited on
Commit
315936d
·
verified ·
1 Parent(s): 895c5d8

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +53 -47
config.json CHANGED
@@ -1,47 +1,53 @@
1
- {
2
- "model": {
3
- "type": "YOLO11n",
4
- "pretrained": true,
5
- "num_classes": 1,
6
- "input_size": 640
7
- },
8
- "dataset": {
9
- "name": "WIDER FACE",
10
- "train": {
11
- "images": "WIDER_train/images",
12
- "annotations": "WIDER_train/annotations"
13
- },
14
- "val": {
15
- "images": "WIDER_val/images",
16
- "annotations": "WIDER_val/annotations"
17
- }
18
- },
19
- "training": {
20
- "epochs": 225,
21
- "batch_size": 16,
22
- "learning_rate": 0.001,
23
- "optimizer": "Adam",
24
- "momentum": 0.9,
25
- "weight_decay": 0.0005,
26
- "scheduler": {
27
- "type": "StepLR",
28
- "step_size": 30,
29
- "gamma": 0.1
30
- }
31
- },
32
- "augmentation": {
33
- "horizontal_flip": true,
34
- "vertical_flip": false,
35
- "rotation": 10,
36
- "scale": [0.8, 1.2],
37
- "shear": 2
38
- },
39
- "evaluation": {
40
- "interval": 1,
41
- "metrics": ["mAP", "precision", "recall"]
42
- },
43
- "save": {
44
- "checkpoint_dir": "checkpoints/",
45
- "best_model": "model.pth"
46
- }
47
- }
 
 
 
 
 
 
 
1
+ {
2
+ "model": {
3
+ "model_type": "YOLO11n",
4
+ "pretrained": true,
5
+ "num_classes": 1,
6
+ "input_size": 640
7
+ },
8
+ "dataset": {
9
+ "name": "WIDER FACE",
10
+ "train": {
11
+ "images": "WIDER_train/images",
12
+ "annotations": "WIDER_train/annotations"
13
+ },
14
+ "val": {
15
+ "images": "WIDER_val/images",
16
+ "annotations": "WIDER_val/annotations"
17
+ }
18
+ },
19
+ "id2label": {
20
+ "0": "face"
21
+ },
22
+ "label2id": {
23
+ "face": "0"
24
+ },
25
+ "training": {
26
+ "epochs": 225,
27
+ "batch_size": 16,
28
+ "learning_rate": 0.001,
29
+ "optimizer": "Adam",
30
+ "momentum": 0.9,
31
+ "weight_decay": 0.0005,
32
+ "scheduler": {
33
+ "type": "StepLR",
34
+ "step_size": 30,
35
+ "gamma": 0.1
36
+ }
37
+ },
38
+ "augmentation": {
39
+ "horizontal_flip": true,
40
+ "vertical_flip": false,
41
+ "rotation": 10,
42
+ "scale": [0.8, 1.2],
43
+ "shear": 2
44
+ },
45
+ "evaluation": {
46
+ "interval": 1,
47
+ "metrics": ["mAP", "precision", "recall"]
48
+ },
49
+ "save": {
50
+ "checkpoint_dir": "checkpoints/",
51
+ "best_model": "model.pth"
52
+ }
53
+ }