def test_slice_converter(self):
input_dim = (1, 4, 2)
output_dim = (1, 2, 2)
inputs = [('input', datatypes.Array(*input_dim))]
outputs = [('output', datatypes.Array(*output_dim))]
builder = NeuralNetworkBuilder(inputs, outputs)
builder.add_slice(name='Slice', input_name='input', output_name='output', axis='height', start_index=0,
end_index=-1, stride=1)
model_onnx = convert_coreml(builder.spec)
self.assertTrue(model_onnx is not None)
def test_slice_converter(self):
input_dim = (1, 4, 2)
output_dim = (1, 2, 2)
inputs = [('input', datatypes.Array(*input_dim))]
outputs = [('output', datatypes.Array(*output_dim))]
builder = NeuralNetworkBuilder(inputs, outputs)
builder.add_slice(name='Slice',
input_name='input',
output_name='output',
axis='height',
start_index=0,
end_index=-1,
stride=1)
context = ConvertContext()
node = SliceLayerConverter.convert(
context, builder.spec.neuralNetwork.layers[0], ['input'],
['output'])
self.assertTrue(node is not None)
