Spaces:
Sleeping
Sleeping
SakshiRathi77
commited on
Commit
•
1d94dcd
1
Parent(s):
192a5c6
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
|