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 最佳化的模型。
相關
- 專案
- 專案
- 專案
- 專案
- 專案