如何向預訓練的 caffe model 添加預處理層？

Question

我有一個預先訓練好的圖像分類 model 保存在 caffe 中，model 預計會得到灰度（單通道）圖像。 我想在僅向 model 提供 RGB（三通道）輸入的工具中使用此 model。 無法更改此工具提供圖像的方式，因此我想在輸入層之前添加一個層，僅將輸入轉換為一個通道，這在 caffe 中可能嗎？ 如何？

如果可能的話，我正在尋找一種不需要為 caffe 定義新層的解決方案。

請注意，我有 model 的“.prototxt”和“.weights”文件。

我之前在 tensorflow 中做過類似的事情，但我不知道這在 caffe 中是否可行，也沒有在網上找到太多資料。

Answer 1

您可以添加一個Python層來為您完成。

什么是Python層。 可以在此處找到此類層的示例。