def test_crop_converter(self):
top_crop = 1
bottom_crop = 1
left_crop = 1
right_crop = 1
input_dim = (8, 6, 4)
output_dim = (8, 6 - top_crop - bottom_crop, 4 - left_crop - right_crop)
input = [('input', datatypes.Array(*input_dim))]
output = [('output', datatypes.Array(*output_dim))]
builder = NeuralNetworkBuilder(input, output)
builder.add_crop(name='Crop', left=left_crop, right=right_crop, top=top_crop, bottom=bottom_crop, offset=[0, 0],
input_names=['input'], output_name='output')
model_onnx = convert_coreml(builder.spec)
self.assertTrue(model_onnx is not None)
def test_crop_converter(self):
top_crop = 1
bottom_crop = 1
left_crop = 1
right_crop = 1
input_dim = (8, 6, 4)
output_dim = (8, 6 - top_crop - bottom_crop,
4 - left_crop - right_crop)
input = [('input', datatypes.Array(*input_dim))]
output = [('output', datatypes.Array(*output_dim))]
builder = NeuralNetworkBuilder(input, output)
builder.add_crop(name='Crop',
left=left_crop,
right=right_crop,
top=top_crop,
bottom=bottom_crop,
offset=[0, 0],
input_names='input',
output_name='output')
context = ConvertContext()
node = CropLayerConverter.convert(context,
builder.spec.neuralNetwork.layers[0],
['input'], ['output'])
self.assertTrue(node is not None)
