将输出图像保存在Caffe中

Save output images in Caffe

本文关键字:Caffe 存在 保存 输出 图像      更新时间:2023-10-16

我对使用CAFFE框架非常新。我有一个用于图像的网络。它将图像作为输入并输出图像。Caffe是否有一层我必须包括以保存输出图像?

谢谢。

caffe没有"保存到图像"层。
您可以尝试查看"HDF5Output"层是否可以适用于您:您可以将输出映像作为浮点数ND数组保存到HDF5二进制文件中,然后稍后使用其他应用程序读取。

另外,您可以使用"Python"层为您保存图像。只需在Python中写"保存代码"(例如使用PIL.Image软件包)。