add default input_channel values to configs

ultralytics · Nov 15, 2023 · 0acac83 · 0acac83
1 parent e29b190
commit 0acac83
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/models/yolov5l.yaml b/models/yolov5l.yaml
@@ -1,6 +1,7 @@
 # YOLOv5 🚀 by Ultralytics, AGPL-3.0 license
 
 # Parameters
+input_channels: 3  # number of input channels, RGB is 3
 nc: 80  # number of classes
 depth_multiple: 1.0  # model depth multiple
 width_multiple: 1.0  # layer channel multiple

diff --git a/models/yolov5m.yaml b/models/yolov5m.yaml
@@ -1,6 +1,7 @@
 # YOLOv5 🚀 by Ultralytics, AGPL-3.0 license
 
 # Parameters
+input_channels: 3  # number of input channels, RGB is 3
 nc: 80  # number of classes
 depth_multiple: 0.67  # model depth multiple
 width_multiple: 0.75  # layer channel multiple

diff --git a/models/yolov5n.yaml b/models/yolov5n.yaml
@@ -1,6 +1,7 @@
 # YOLOv5 🚀 by Ultralytics, AGPL-3.0 license
 
 # Parameters
+input_channels: 3  # number of input channels, RGB is 3
 nc: 80  # number of classes
 depth_multiple: 0.33  # model depth multiple
 width_multiple: 0.25  # layer channel multiple

diff --git a/models/yolov5s.yaml b/models/yolov5s.yaml
@@ -1,6 +1,7 @@
 # YOLOv5 🚀 by Ultralytics, AGPL-3.0 license
 
 # Parameters
+input_channels: 3  # number of input channels, RGB is 3
 nc: 80  # number of classes
 depth_multiple: 0.33  # model depth multiple
 width_multiple: 0.50  # layer channel multiple