PINTO_model_zoo: 一个用于边缘设备跨框架部署的预转换和预量化模型集合
PINTO_model_zoo: 一个用于边缘设备跨框架部署的预转换和预量化模型集合
它解决了什么问题
PINTO_model_zoo 提供了一个集中化的预转换和预量化 AI 模型集合。它通过提供多种格式的常用模型的即用版本,解决了在不同深度学习框架之间手动转换模型以及为边缘设备(如 Raspberry Pi 或 EdgeTPU)优化模型的困难。
工作原理
该项目维护了一个在各种框架之间进行过相互转换的模型“动物园”(zoo)。它支持广泛的格式,包括 TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlow Lite (Float32, Float16, 和 INT8), EdgeTPU, 和 CoreML。该仓库包含用于图像分类、2D 对象检测和其他视觉任务的模型,通常提供多种量化级别(Weight, Integer, Full Integer, 和 Dynamic Range)以平衡性能和精度。
适用人群
从事边缘 AI 和移动端部署的开发者和研究人员,他们需要特定格式的优化模型,而无需亲自执行复杂的转换和量化过程。
亮点
- 广泛的框架支持:支持几乎所有主要的 AI 框架和部署格式(TFLite, ONNX, CoreML, OpenVINO 等)。
- 丰富的模型库:包含大量用于图像分类和对象检测的预量化模型。
- 边缘优化:专门针对 Raspberry Pi 4/3 和 EdgeTPU 等硬件的性能进行优化。
- 量化多样性:提供 FP32, FP16, INT8, 和 Dynamic Range 量化模型,以适应不同的硬件限制。
Sources
- undefinedPINTO0309/PINTO_model_zoo