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