Skip to content
Advertisement

Tag: pycaffe

How to add a preprocessing layer to a pretrained caffe model?

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

Advertisement