turicreate.style_transfer.StyleTransfer.export_coreml

StyleTransfer.export_coreml(filename, image_shape=(256, 256), include_flexible_shape=True)

Save the model in Core ML format. The Core ML model takes an image of fixed size, and a style index inputs and produces an output of an image of fixed size

Parameters:
path : string

A string to the path for saving the Core ML model.

image_shape: tuple

A tuple (defaults to (256, 256)) will bind the coreml model to a fixed shape.

include_flexible_shape: bool

Allows the size of the input image to be flexible. Any input image were the height and width are at least 64 will be accepted by the Core ML Model.

See also

save

Examples

>>> model.export_coreml('StyleTransfer.mlmodel')