Я очень новичок в использовании Caffe Framework. У меня есть сеть, которая работает с изображениями. Он принимает изображения в качестве входных и выходных изображений. Есть ли в caffe слой, который я должен включить, чтобы сохранить выходные изображения?
Спасибо.
У Caffe нет слоя «сохранить изображение».
Вы можете попробовать и посмотреть, если "HDF5Output"
Слой может работать на вас: вы можете сохранить свое выходное изображение в виде массива ND с плавающей точкой в двоичный файл hdf5 и прочитать его позже с другими приложениями.
Кроме того, вы можете использовать "Python"
слой, чтобы сохранить изображения для вас. Просто напишите код сохранения на python (используя PIL.Image
пакет например).
Других решений пока нет …