如何向预训练的 caffe model 添加预处理层？

Question

I have a pre-trained image classification model saved in caffe, the model is expected to get grayscale(one channel) images. I want to use this model in a tool that only provides input of RGB(three channels) to the model. It is not possible to change the way this tool provides images so I thought of adding a layer before the input layer that transforms the input to one channel only, is that possible in caffe? and how?

I'm looking for a solution that doesn't require to define new layers to caffe if possible.

Note that I have the ".prototxt" and the ".weights" files of the model.

I previously did a similar thing in tensorflow but I don't know if this is possible in caffe and didn't find much material online.

Answer 1

You can add a Python layer to do it for you.

What is a Python layer . An example of such a layer can be found here .