search

hhk7734 / tensorflow-yolov4

YOLOv4 Implemented in Tensorflow 2.
MIT License
136 stars 75 forks source link

Grayscale prediction #60

Open pmikas opened 3 years ago

pmikas commented 3 years ago

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

hhk7734 commented 3 years ago

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.

pmikas commented 3 years ago

Thanks for the reply

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

hhk7734 commented 3 years ago

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

hhk7734 commented 3 years ago

Can you share 1 channel dataset?