espressif/esp-dl

Espressif deep-learning library for AIoT applications

解決的問題

ESP-DL 是專為在資源受限的 ESP 系列晶片上部署神經網路模型而設計。它提供一個輕量級的推論框架,讓開發者能直接在裝置上執行 AI 應用,無需為基本 AI 任務持續連接雲端。

工作原理

此框架使用基於 FlatBuffers 的自訂輕量級模型格式(.espdl),實現零複製反序列化。它與 ESP-PPQ 量化工具整合,可將 ONNX、PyTorch 和 TensorFlow 格式的模型轉換為最佳化的 ESP-DL 格式。為在 ESP32 系列晶片上最大化效能,採用多項硬體特定最佳化:

  • 靜態記憶體規劃器:根據可用內部 RAM,自動將各層分配至最佳記憶體位置。
  • 雙核心排程:將計算密集型運算子(如 Conv2D)分散至雙核心處理器,以提升速度。
  • 8 位元 LUT 活化:使用查表法(LUT)實現活化函數,降低計算複雜度。
  • 高效運算子:實作 Conv、Gemm、Add 和 Mul 等常見 AI 運算子的優化版本。

適用對象

使用 Espressif 系統單晶片(SoC)開發 AI 驅動的物聯網裝置的嵌入式開發者與 AI 工程師。

主要亮點

  • 廣泛模型支援:支援 YOLO11n、YOLO26 和 PP-OCRv6 等模型的量化與部署。
  • 無縫整合:可直接與 ESP-IDF 整合,並與 ESP-PPQ 量化工具協同運作。
  • 硬體最佳化:專為 ESP32-S3 和 ESP32-P4 晶片優化,達成最大效率。
  • AI Agent 工具鏈:提供工具協助 AI Agent 自動實作並最佳化框架中的運算子。

相關

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