JhonnyChems commited on
Commit
aa4d97e
verified
1 Parent(s): d5e6b46

Upload config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.yaml +104 -0
config.yaml ADDED
@@ -0,0 +1,104 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # ContourSight - Configuraci贸n por Defecto
2
+ # =========================================
3
+ # Precise Mosaic Detector (PMD) Configuration
4
+
5
+ # Configuraci贸n del Modelo
6
+ model:
7
+ type: "unet" # Opciones: unet, unet_tiny, yolo
8
+ in_channels: 3
9
+ out_channels: 1
10
+ base_features: 64
11
+ use_attention: true # Atenci贸n para mejor precisi贸n de contornos
12
+
13
+ # Configuraci贸n espec铆fica de YOLOv8 (si type = yolo)
14
+ yolo:
15
+ model_size: "n" # n, s, m, l, x
16
+ pretrained: true
17
+
18
+ # Configuraci贸n de Entrenamiento
19
+ training:
20
+ epochs: 100
21
+ batch_size: 8
22
+ learning_rate: 0.0001
23
+ weight_decay: 0.0001
24
+
25
+ # Mixed Precision Training (FP16)
26
+ use_amp: true
27
+
28
+ # Gradient Clipping
29
+ gradient_clip: 1.0
30
+
31
+ # Learning Rate Scheduler
32
+ scheduler:
33
+ type: "cosine" # Opciones: cosine, step, plateau
34
+ warmup_epochs: 5
35
+ min_lr: 0.000001
36
+
37
+ # Early Stopping
38
+ early_stopping:
39
+ enabled: true
40
+ patience: 15
41
+ min_delta: 0.0001
42
+
43
+ # Configuraci贸n de P茅rdidas
44
+ loss:
45
+ type: "combined" # Opciones: dice, iou, boundary, focal, combined
46
+
47
+ # Pesos para CombinedLoss
48
+ weights:
49
+ dice: 1.0
50
+ iou: 1.0
51
+ boundary: 2.0 # Mayor peso para precisi贸n de contornos (位)
52
+ focal: 0.5
53
+
54
+ # Par谩metros espec铆ficos
55
+ boundary:
56
+ theta0: 3.0
57
+ theta: 5.0
58
+ use_distance_transform: true
59
+
60
+ # Configuraci贸n de Datos
61
+ data:
62
+ image_size: [512, 512]
63
+ train_split: 0.8
64
+ val_split: 0.1
65
+ test_split: 0.1
66
+
67
+ # Rutas (relativas al directorio ra铆z del proyecto)
68
+ train_images: "data/train/images"
69
+ train_masks: "data/train/masks"
70
+ val_images: "data/val/images"
71
+ val_masks: "data/val/masks"
72
+
73
+ # Augmentations
74
+ augmentation:
75
+ enabled: true
76
+ horizontal_flip: 0.5
77
+ vertical_flip: 0.3
78
+ rotation: 0.3
79
+ brightness_contrast: 0.3
80
+ elastic_transform: 0.2
81
+
82
+ # Configuraci贸n del DataLoader
83
+ dataloader:
84
+ num_workers: 4
85
+ pin_memory: true
86
+ prefetch_factor: 2
87
+
88
+ # Logging y Checkpoints
89
+ logging:
90
+ checkpoint_dir: "models/checkpoints"
91
+ tensorboard_dir: "runs"
92
+ log_every: 10
93
+ save_every_epoch: 5
94
+ wandb:
95
+ enabled: false
96
+ project: "contoursight"
97
+ entity: null
98
+
99
+ # M茅tricas objetivo
100
+ targets:
101
+ contour_accuracy: 98.0 # Objetivo principal: >98%
102
+ iou: 0.95
103
+ dice: 0.97
104
+ boundary_iou: 0.90