Spaces:
Sleeping
Sleeping
Create config.yaml
Browse files- config.yaml +57 -0
config.yaml
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# YOLOv9 Configuration File
|
| 2 |
+
|
| 3 |
+
# Model architecture details
|
| 4 |
+
model:
|
| 5 |
+
type: YOLOv9 # Model type
|
| 6 |
+
backbone:
|
| 7 |
+
type: CSPDarknet # Backbone architecture type
|
| 8 |
+
depth_multiple: 1.0 # Depth multiplier
|
| 9 |
+
width_multiple: 1.0 # Width multiplier
|
| 10 |
+
|
| 11 |
+
# Input image settings
|
| 12 |
+
input_size: 640 # Input image size (square)
|
| 13 |
+
|
| 14 |
+
# Anchor parameters
|
| 15 |
+
anchors:
|
| 16 |
+
- [10, 13, 16, 30, 33, 23] # Anchor box 1
|
| 17 |
+
- [30, 61, 62, 45, 59, 119] # Anchor box 2
|
| 18 |
+
- [116, 90, 156, 198, 373, 326] # Anchor box 3
|
| 19 |
+
|
| 20 |
+
# Training hyperparameters
|
| 21 |
+
hyp:
|
| 22 |
+
lr0: 0.01 # Initial learning rate
|
| 23 |
+
lrf: 0.01 # Learning rate reduction factor
|
| 24 |
+
momentum: 0.937 # Momentum
|
| 25 |
+
weight_decay: 0.0005 # Weight decay
|
| 26 |
+
warmup_epochs: 3.0 # Warmup epochs
|
| 27 |
+
warmup_momentum: 0.8 # Warmup momentum
|
| 28 |
+
warmup_bias_lr: 0.1 # Warmup bias learning rate
|
| 29 |
+
box: 7.5 # Box loss gain
|
| 30 |
+
cls: 0.5 # Class loss gain
|
| 31 |
+
cls_pw: 1.0 # Class label smoothing
|
| 32 |
+
dfl: 1.5 # DFL loss gain
|
| 33 |
+
obj_pw: 1.0 # Objectness loss gain
|
| 34 |
+
iou_t: 0.2 # IoU training threshold
|
| 35 |
+
anchor_t: 5.0 # Anchor matching threshold
|
| 36 |
+
fl_gamma: 0.0 # Focal loss gamma
|
| 37 |
+
hsv_h: 0.015 # HSV hue gain
|
| 38 |
+
hsv_s: 0.7 # HSV saturation gain
|
| 39 |
+
hsv_v: 0.4 # HSV value gain
|
| 40 |
+
degrees: 0.0 # Image rotation (degrees)
|
| 41 |
+
translate: 0.1 # Image translation
|
| 42 |
+
scale: 0.9 # Image scale
|
| 43 |
+
shear: 0.0 # Image shear
|
| 44 |
+
perspective: 0.0 # Image perspective transform
|
| 45 |
+
flipud: 0.0 # Flip image vertically
|
| 46 |
+
fliplr: 0.5 # Flip image horizontally
|
| 47 |
+
mosaic: 1.0 # Mosaic augmentation
|
| 48 |
+
mixup: 0.15 # Mixup augmentation
|
| 49 |
+
copy_paste: 0.3 # Copy-Paste augmentation
|
| 50 |
+
|
| 51 |
+
# Other settings
|
| 52 |
+
other:
|
| 53 |
+
multi_scale: false # Use multi-scale training
|
| 54 |
+
flip: true # Use random horizontal flipping
|
| 55 |
+
blur: false # Use random image blurring
|
| 56 |
+
letterbox: false # Use letterbox resizing
|
| 57 |
+
rect: false # Use rectangular resizing
|