slice/split eine Schicht in Keras wie in caffe

Question

Ich habe this Konverter verwendet, um ein Caffe-Modell zu Keras zu konvertieren. Aber einer meiner Layer ist vom Typ slice und muss ebenfalls konvertiert werden, aber der Konverter unterstützt dies derzeit nicht und löst eine Ausnahme aus. Gibt es Arbeit dafür? Hier ist meine Schicht:

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

Answer 1

Es scheint, dass Sie eine Lambda Schicht verwendet werden soll. In diesem Fall kann man das folgendermaßen:

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

Beachten Sie, dass in x Sie die Proben Achse berücksichtigen müssen, während in output_shape ist es nicht mehr benötigt.