YOLO Detection on Grayscale Images

Hi,

I am trying to detect objects in the grayscale images, but I get the following error.

"The input image channel size (1) must be the same as the network's input channel size (3)."

Per documentation, it says the network accpets single channel as well, but what could be the reason for the error?

Thanks!

[Sunil Patidar]

Hi,

Which MATLAB are you using?

[Kev]

MATLAB 2020a, and same issue was observed on 2020b.

Answer 1

[Sunil Patidar]

Hi Kev,

It seems to be an issue with the 'trainYOLOV2ObjectDetector' function not being able to accept images that are of size [M x N x 1]. As a workaround you might need to duplicate your grayscale image across all three channels, and use a network that accepts [M x N x 3] sized matrices.

[Kev]

It works after converting single channel grayscale into 3 channel grayscale.