K-文化的一切——从回归到 K-美妆,发送到您的邮箱订阅邮件

METAL MEDIA

google-ai-edge/models-samples

25Jupyter NotebookApache-2.0

把PyTorch训练的AI模型转换成手机可用格式的示例代码

这个仓库隶属于谷歌的AI Edge项目,收集了模型转换和制作相关的示例代码。其中convert_pytorch目录下放着几个Colab笔记本,演示如何下载PyTorch模型并将其转换成能在手机等设备上运行的TensorFlow Lite格式。只要有谷歌账号,就能直接在云端浏览器里运行这些笔记本,不需要在本地安装任何环境。项目获得25个star,主要语言是Jupyter Notebook。

它做什么

  1. 该仓库由google-ai-edge维护,汇集了创建和转换AI模型的示例。
  2. convert_pytorch目录提供将PyTorch模型转换为TensorFlow Lite格式的Colab笔记本。
  3. 笔记本运行在谷歌云端,只需个人谷歌账号即可运行,无需本地安装。
  4. 面向想把AI模型部署到手机等设备端(on-device)环境的开发者。

为什么重要

想让AI模型直接跑在手机或小型设备上,通常需要先做格式转换,这个仓库提供了可以实际动手操作的代码。对于喜欢边跑代码边学习、而不是只看理论的开发者来说,这是一个实用的入门材料。

本仓库术语

  • PyTorch · Meta推出的常用AI模型开发工具
  • TensorFlow Lite · 谷歌推出的、用于在手机等性能有限设备上高效运行AI模型的格式
  • Colab · 谷歌提供的免费云端笔记本服务,可在浏览器中直接运行代码
  • AI Edge · 谷歌关于在用户设备本地而非服务器上直接运行AI的技术方向

仓库简介(英文)

google-ai-edge/models-samples

所属 · google-deepmind

在 GitHub 打开

热门仓库

全部仓库 →

METAL MEDIA 最新报道