2017-03-27 13 views
2

私はthisコンバータを使用して、CaffeモデルをKerasに変換しました。しかし、私の層の1つはタイプsliceであり、変換する必要がありますが、コンバータは現在これをサポートしておらず、例外が発生しています。それのための回避策はありますか?ここに自分のレイヤーがあります:ケーキのようにケラスのレイヤーを分割/分割する

layer { 
    name: "slice_label" 
    type: SLICE 
    bottom: "label" 
    top: "label_wpqr" 
    top: "label_xyz" 
    slice_param { 
     slice_dim: 1 
     slice_point: 4 
    } 
} 

答えて

2

Lambdaレイヤーを使いたいと思うようです。 xにあなたがoutput_shapeにそれはもはや必要ではないです一方のアカウントにサンプル軸を取る必要があることを

sliced = Lambda(lambda x: x[:,slicing_indeces], output_shape=(sliced_shape))(input) 

注:この場合は次の操作を行うことがあります。

+0

私の答えはあなたを助けましたか? –

関連する問題