Grayscale images

[AlexisPapaioannou]

❔Question

Hello,

When I used the Yolov5 with gray scale images I received the below error. I had also added a parameter ch: 1 into the yaml file.

Additional context

Given groups=1, weight of size [8, 1, 3, 3], expected input[16, 3, 128, 128] to have 1 channels, but got 3 channels instead

The custom yaml file

parameters

nc: {num_classes} # number of classes ch: 1 # input channels depth_multiple: 0.33 # model depth multiple width_multiple: 0.35 # layer channel multiple

anchors

anchors:

• [4,5, 8,10, 13,16] # P3/8
• [23,29, 43,55, 73,105] # P4/16
• [146,217, 231,300, 335,433] # P5/32

YOLOv5 backbone

[from, number, module, args] # kernel, padding, stride

backbone:

[from, number, module, args]

[[-1, 1, Conv, [16, 3, 2]], # 0-P1/2 [-1, 1, Conv, [8, 3, 2]], # 1-P2/4 [-1, 1, Conv, [32, 3, 2]], # 1-P2/4 [-1, 3, C3, [32]], [-1, 1, Conv, [32, 3, 2]], # 3-P3/8 [-1, 9, C3, [64]], [-1, 1, Conv, [64, 3, 2]], # 5-P4/16 [-1, 9, C3, [128]], [-1, 1, Conv, [128, 3, 2]], # 7-P5/32 [-1, 1, SPP, [256, [5, 9, 13]]], [-1, 3, C3, [128, False]], # 9 ]

[glenn-jocher]

@AlexisPapaioannou no modifications to anything are required to train greyscale.

[github-actions[bot]]

👋 Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 🚀 resources:

• Wiki – https://github.com/ultralytics/yolov5/wiki
• Tutorials – https://docs.ultralytics.com/yolov5
• Docs – https://docs.ultralytics.com

Access additional Ultralytics ⚡ resources:

• Ultralytics HUB – https://ultralytics.com/hub
• Vision API – https://ultralytics.com/yolov5
• About Us – https://ultralytics.com/about
• Join Our Team – https://ultralytics.com/work
• Contact Us – https://ultralytics.com/contact

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐!