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 测试该库(例如用于鸢尾花分类器)。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目