YOLOConfiguration does not support number of channels!=3

[kfirgedal]

Hi,

YOLOConfiguration implementation does not allow you to provide a number of channels in the .cfg file that is different than 3.

The line that is causing the problem is:
https://github.com/PyTorchLightning/lightning-bolts/blob/2415b49a2b405693cd499e09162c89f807abbdc4/pl_bolts/models/detection/yolo/yolo_config.py#L43

Thanks

[senarvi]

That's right, the number of channels in the input image is always 3 (RGB). @kfirgedal is there a way to specify the number of channels in the Darknet implementation?

[kfirgedal]

@senarvi Yes.
Number of channels is defined in the configuration file:
https://github.com/AlexeyAB/darknet/blob/b4d03f88028594c3073c41f60cd469d03fb3ee8e/cfg/yolov4-csp.cfg#L10
It is being parsed when the network is constructed:
https://github.com/AlexeyAB/darknet/blob/b4d03f88028594c3073c41f60cd469d03fb3ee8e/src/parser.c#L1210

[senarvi]

@kfirgedal indeed you're right. It should be trivial to add support for that. I can take a look at it soon, unless you want to create a pull request. I'm currently working on a big update that would bring features from YOLOv5, YOLOX, and Scaled-YOLOv4.

[kfirgedal]

@senarvi
#806

[senarvi]

@senarvi #806

Oh you already had a pull request. Excellent! I added a small suggestion, but I'm happy with this being merged.

[stale]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.