Create config.yaml

config.yaml

@@ -0,0 +1,30 @@
+model:
+  name: NOAA_AFSC_MML_Iceseals_31K
+  architecture: YOLOv11n
+  nc: 8  # Number of classes
+  img_size: 1024
+  batch_size: 113
+  optimizer: SGD
+  lr: 0.01
+  momentum: 0.9
+  epochs: 64
+
+dataset:
+  train: "path/to/train"
+  val: "path/to/val"
+  test: "path/to/test"
+  classes:
+    - bearded_pup
+    - bearded_seal
+    - ribbon_pup
+    - ribbon_seal
+    - ringed_pup
+    - ringed_seal
+    - spotted_pup
+    - spotted_seal
+
+augmentations:
+  - Blur: {p: 0.01, blur_limit: [3, 7]}
+  - MedianBlur: {p: 0.01, blur_limit: [3, 7]}
+  - ToGray: {p: 0.01, num_output_channels: 3, method: "weighted_average"}
+  - CLAHE: {p: 0.01, clip_limit: [1.0, 4.0], tile_grid_size: [8, 8]}