前言:
這一兩天我一直很想把.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)