tensorflow-yolov4 icon indicating copy to clipboard operation
tensorflow-yolov4 copied to clipboard

Published 20 hours ago • verified publisher icon hhk7734

Grayscale prediction

Open pmikas opened this issue 4 years ago • 4 comments

Hello!

I trained a model in grayscale, so I am also trying to predict in grayscale. Is there a way to set the prediction with 1 channel?

Regards

pmikas avatar Feb 10 '21 12:02 pmikas

https://github.com/hhk7734/tensorflow-yolov4/blob/4f75970d02b60960257f9f7baa091e0e6874a890/py_src/yolov4/tf/init.py#L55-L84

Line 66 - inputs = layers.Input([self.input_size[1], self.input_size[0], 3]) try to change 3 to 1.

hhk7734 avatar Feb 10 '21 13:02 hhk7734

Thanks for the reply

It doesn't work. Even adjusting other parts of the code, the output isn't the same as from darknet

pmikas avatar Feb 11 '21 17:02 pmikas

I'm working on yolov4 v3.0. I will try after release.

hhk7734 avatar Feb 11 '21 22:02 hhk7734

Can you share 1 channel dataset?

hhk7734 avatar Feb 13 '21 12:02 hhk7734