像在caffe中一样切片/分割keras中的图层

Question

我用这个转换器将Caffe模型转换为Keras.但是我的一个层是类型的slice,它也需要转换,但转换器目前不支持这个并引发异常.它有什么工作吗？这是我的图层:

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

Answer 1

看来你想要使用一个Lambda图层.在这种情况下,您可以执行以下操作:

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

请注意,x您需要考虑样本轴,而output_shape不再需要它.