PINTO_model_zoo
499个CV模型的跨框架转换仓库,MIT许可,覆盖TensorFlow/PyTorch/ONNX/OpenVINO/TFLite/EdgeTPU/CoreML全框架
加载项目详情…
本应用为开源项目,仅供学习研究,请遵守其开源协议。
499个CV模型的跨框架转换仓库,MIT许可,覆盖TensorFlow/PyTorch/ONNX/OpenVINO/TFLite/EdgeTPU/CoreML全框架
加载项目详情…
本应用为开源项目,仅供学习研究,请遵守其开源协议。
想象一下:你手里有一把瑞士军刀,能把任意框架训练的 AI 模型「变形」成任何其他框架都能用的格式——TensorFlow、PyTorch、ONNX、OpenVINO、TensorFlow Lite、EdgeTPU、CoreML……这把瑞士军刀,就是日本开发者 Katsuya Hyodo(GitHub @PINTO0309)花费数年维护的 PINTO_model_zoo。

项目封面图:PINTO_model_zoo 汇聚多框架模型转换能力
深度学习江湖,框架割据已久。
TensorFlow 开发者想用 PyTorch 训练好的模型?难。PyTorch 团队想在手机端跑 ONNX 模型?得绕路。Intel 想把模型部署到 CPU?得找专门的转换工具。每个框架都有自己的「方言」,模型文件格式互不相通——这被从业者形象地称为 「框架孤岛」问题。
日本开发者 Katsuya Hyodo 以个人之力,花费数年时间,逐一破解了这些壁垒。他不仅自己做转换,还把 499 个经过他亲手调优的转换结果,全部开源上传到这个仓库里。仓库于 2020 年左右开始维护,至今已积累超过 4568 颗星、3841 次提交,成为计算机视觉领域极具影响力的个人贡献。
PINTO_model_zoo 的核心价值,用一句话概括就是:一个仓库,收录了数百个经过验证的跨框架转换模型。
每个模型子目录(如 001_deeplabv3、018_EfficientDet、227_face-detection-adas-0001)通常包含:
download.sh — 一键下载预训练权重*.py — Python 推理/演示脚本saved_model/、.tflite、.onnx、.xml/.bin(OpenVINO IR)等多格式文件支持的转换框架矩阵:TensorFlow、PyTorch、ONNX、OpenVINO、TF-Lite(Float32/Float16/INT8/UINT8)、EdgeTPU、CoreML、TF.js、TF-TRT,覆盖从高精度到极致轻量化的全部需求。
PINTO0309 维护了另一把「瑞士军刀」——openvino2tensorflow,这是本仓库背后的核心转换引擎。工作流程如下:
PyTorch 模型 (.pt)
→ ONNX 中间格式
→ OpenVINO IR (.xml/.bin)
→ TensorFlow SavedModel / .h5 / .tflite / .pb
→ EdgeTPU 量化模型
过程中涉及大量细节处理:张量维度顺序(NCHW→NHWC)、算子替换、量化感知训练(QAT)等。仓库中每个模型子目录的转换脚本,就是这套流程的具体实例。
PINTO_model_zoo 中的模型覆盖了计算机视觉的主流任务:
社区贡献者一览,项目持续活跃
这不是一个「一键运行」的项目,而是一个模型资源的精选集。上手方式取决于你的具体需求:
方式一:下载现成模型(最常用)
git clone https://github.com/PINTO0309/PINTO_model_zoo.git
cd PINTO_model_zoo/002_mobilenetv3-ssd
./download.sh # 下载权重
python3 mobilenetv3ssd.py # 运行演示
方式二:自己转换新模型 需要安装 openvino2tensorflow 及相关依赖(TensorFlow、PyTorch、ONNX、OpenVINO Toolkit 等),按仓库中各子目录的脚本模板操作。
硬件建议:
License 复杂:转换脚本本身是 MIT License,但上游模型各有自己的 License,使用前必须逐目录查阅 LICENSE 文件。
部分模型已过时:MobileNetV3 量化模型和全整数量化模型在某些版本存在问题,遇到相关模型需仔细测试。
非统一部署:499 个模型子目录各自独立,没有统一的容器化方案,不提供 Web UI。生产级部署仍需自行封装。
文档依赖日语:作者的主要博客在 Qiita(日语技术社区),部分高质量教程仅有日语版本。
PINTO_model_zoo 的意义远超一个模型仓库本身。它代表了一种独特的开源贡献模式:以个人之力,长期填补框架间互操作性的空白。
在 AI 行业,大厂往往只维护自家框架的模型 zoo(TensorFlow Model Garden、PyTorch Hub、Intel OpenVINO Model Zoo),而跨框架的「最后一公里」往往无人问津。PINTO0309 填补了这个空白,让边缘设备部署(EdgeTPU、CoreML、TFLite)变得触手可及。
项目自 2020 年起持续活跃,2025 年仍有关于 MediaPipe Pose 模型更新的 Issue 维护。Zenodo 提供了正式 DOI(10.5281/zenodo.10229410),可用于学术引用。
一句话总结:如果你需要把一个计算机视觉模型部署到边缘设备(手机、树莓派、Coral、苹果设备),PINTO_model_zoo 几乎是你能绕开的最佳起点。