1. 程式人生 > >caffe 模型轉換(不同深度學習框架互轉)

caffe 模型轉換(不同深度學習框架互轉)

本人使用的是微軟的MMdnn工具,具體使用方法見其GitHub工程,這裡說明下在轉換過程中的問題:

1、由caffe轉mxnet時,在卷積層和FC層之間,mmconvert轉換工具會自動增加一個Flatten層,用於對接卷積層和FC層。這將導致轉換後的FC層與caffe定義的FC層的名字不同(如caffe中FC層名為fc1,則轉換後flatten層的名字為fc1_0,FC層名字為fc1_1,即自動加了字尾)。所以注意在定義caffe層時,養成在卷積層和FC層加Flatten層的習慣(雖然預設不加也可以)。