PINTO0309/PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

解決的問題

在不同軟體框架(例如 PyTorch 轉 TensorFlow)之間轉換 AI 模型,並針對特定硬體(例如手機或 Edge TPU)進行優化,通常是一個複雜且需手動處理的過程。本專案提供一組預先轉換與量化好的模型,讓開發者無需自行完成這些工作。

運作方式

此程式庫扮演「模型動物園」的角色,儲存了跨多種格式相互轉換的廣泛神經網路模型。支援廣泛的框架與執行環境,包括:

  • TensorFlow / TensorFlow Lite(含 Float32、Float16 與 INT8 量化)
  • PyTorch
  • ONNX
  • OpenVINO
  • CoreML
  • TFJS(TensorFlow.js)
  • EdgeTPU

適用對象

需要將現有 AI 模型部署至邊緣裝置、行動應用程式或特定硬體加速器,但不願自行撰寫轉換與量化腳本的開發者與研究人員。

主要特色

  • 廣泛的格式支援:模型以 .tflite.h5.pb.onnx.mlmodel 等格式提供。
  • 多樣化量化選項:提供不同量化層級(權重、整數、全整數、Float16)的模型,以平衡效能與準確度。
  • 多元模型圖書館:包含大量用於影像分類與 2D 物件偵測的模型(例如 EfficientNet、MobileNet、YOLO 變體)。
  • 硬體最佳化:特別包含針對 EdgeTPU 與 Raspberry Pi 最佳化的模型。

相關

  • 專案
  • 專案
  • 專案
  • 專案
  • 專案