iT邦幫忙

2021 iThome 鐵人賽

DAY 20
0
自我挑戰組

IOS、Python自學心得30天系列 第 20

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

  • 分享至 

  • xImage
  •  

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

上一篇
IOS、Python自學心得30天 Day-19 TensorFlowLiteSwift模組
下一篇
IOS、Python自學心得30天 Day-21 CoreML範例
系列文
IOS、Python自學心得30天30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言