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)