Load and convert the tensorflow model import coremltools as ct
model = ct.convert( "/path/to/saved_model", # Path to the SavedModel directory inputs=[ ct.ImageType(shape=(1, 256, 256, 3), name="contentImage", scale=1/255), ct.ImageType(shape=(1, 256, 256, 3), name="styleImage", scale=1/255) ], outputs=[ct.TensorType(name="stylizedImage")], source="tensorflow" )
model.save("DualInputStylization.mlmodel"