IOS、Python自学心得30天 Day-20 .h5 to .tflite to .mlmodel

前言:
这一两天我一直很想把.h5档案或是.pb档案
直接转成Xcode可用的.mlmodel档案
但我目前还没找到方法
途中发现可以转成TensorFlowLite的.tflite档案
安卓系统的话应该就能使用到
但能不能应用在ios还需要些时间

import tensorflow as tf

# Convert the model
converter = tf.lite.TFLiteConverter.from_saved_model('Xcode.mlmodel') # path to the SavedModel directory
tflite_model = converter.convert()

# Save the model.
with open('model.tflite', 'wb') as f:
  f.write(tflite_model)