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

Question

我使用this转换器将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中则不再需要它。