added changes to use grayscale images during training

2023-08-21 10:51:22 +02:00
parent e8dcc30754
commit 833d864689
3 changed files with 10 additions and 3 deletions
--- a/ultralytics/cfg/models/v8/yolov8.yaml
+++ b/ultralytics/cfg/models/v8/yolov8.yaml
@ -3,6 +3,7 @@

 # Parameters
 nc: 80  # number of classes
+ch: 1  # number of channels
 scales: # model compound scaling constants, i.e. 'model=yolov8n.yaml' will call yolov8.yaml with scale 'n'
  # [depth, width, max_channels]
  n: [0.33, 0.25, 1024]  # YOLOv8n summary: 225 layers,  3157200 parameters,  3157184 gradients,   8.9 GFLOPs