將輸出圖像保存在Caffe中

Question

我對使用caffe框架非常陌生。我有一個在圖像上工作的網絡。它將圖像作爲輸入和輸出圖像。咖啡有一層我必須包括，以保存輸出圖像？

謝謝。

Answer 1

Caffe沒有「保存到圖像」圖層。
您可以嘗試看看"HDF5Output"圖層是否可以爲您工作：您可以將輸出圖像作爲浮點ND數組保存到hdf5二進制文件中，並在稍後與其他應用程序一起讀取。可以使用"Python"圖層爲您保存圖像。只需在python中編寫保存代碼（例如使用PIL.Image包）。