在Caffe中保存输出图像

Question

我是使用caffe框架的新手。我有一个可以处理图像的网络。它将图像作为输入和输出图像。 caffe是否有一个我必须包含的图层以保存输出图像？

谢谢。

Answer 1

Caffe没有“保存到图像”图层。 您可以尝试查看"HDF5Output"图层是否适合您：您可以将输出图像作为浮点ND数组保存到hdf5二进制文件中，稍后可以与其他应用程序一起阅读。

或者，您可以使用"Python"图层为您保存图像。只需在python中编写保存代码（例如使用PIL.Image包）。