将输出图像保存在Caffe中

Question

我对使用caffe框架非常陌生。我有一个在图像上工作的网络。它将图像作为输入和输出图像。咖啡有一层我必须包括，以保存输出图像？

谢谢。

Answer 1

Caffe没有“保存到图像”图层。
您可以尝试看看"HDF5Output"图层是否可以为您工作：您可以将输出图像作为浮点ND数组保存到hdf5二进制文件中，并在稍后与其他应用程序一起读取。可以使用"Python"图层为您保存图像。只需在python中编写保存代码（例如使用PIL.Image包）。