coremltools_convert_keras_mobilenetv2_to_mlpackage.py

1import coremltools as ct
2import tf_keras as keras
3
4# 1. Load a pre-trained model
5keras_model = keras.applications.MobileNetV2(
6    weights="imagenet", 
7    input_shape=(224, 224, 3)
8)
9
10# 2. Convert the model to Core ML
11# We provide the model and define the input type (Image)
12model = ct.convert(
13    keras_model,
14    inputs=[ct.ImageType(shape=(1, 224, 224, 3), scale=1/127.5, bias=[-1])]
15)
16
17# 3. Save the converted model
18model.save("MobileNetV2.mlpackage")
19
20print("Model converted and saved successfully.")