mattn/go-tflite

Go binding for TensorFlow Lite

解決的問題

提供開發者在 Go 程式語言中使用 TensorFlow Lite 模型的方法,彌補那些希望在 Go 應用程式中執行輕量級機器學習模型,卻又不需撰寫 C++ 或 Python 的開發者需求。

運作方式

本專案作為 TensorFlow Lite C API 的 Go 綁定。允許 Go 程式載入 .tflite 模型檔案,設定解釋器選項,配置張量,並透過傳遞輸入資料給模型及取得輸出結果來執行推論。

適用對象

需要將預訓練的 TensorFlow Lite 模型整合至軟體中的 Go 開發者,包括針對邊緣裝置或使用 Edge TPU 等硬體加速器的開發者。

主要特色

  • TFLite 整合:直接綁定 TensorFlow Lite C API,可在 Go 中執行推論。
  • Edge TPU 支援:與 Edge TPU 代理相容,支援硬體加速機器學習。
  • 瀏覽器測試:提供 Binder 連結,可透過 Jupyter Notebook 測試該函式庫(例如用於鳶尾花分類器)。

相關

  • 專案
  • 專案
  • 專案
  • 專案
  • 專案