簡體   English   中英

在 EDGE TPU 上加速多模型推理

[英]Speed Up Multiple Model Inference on EDGE TPU

我已經重新訓練了一個 RESNET50 模型,用於在 EDGE TPU 上重新識別。 但是,似乎沒有辦法將一批圖像提取到EDGE_TPU。

我想出了一個為圖像運行多個相同模型的解決方案。

但是,無論如何可以加快多個模型的模型推理? 現在的線程甚至比單模型推理還要慢

是的,edgetpu 的架構師不允許批量處理。 您是否嘗試過模型流水線? https://coral.ai/docs/edgetpu/pipeline/

不幸的是,目前僅在 C++ 中可用,但我們希望在第四季度中期將其擴展到 python。

因為批量推理現在不可用,所以流水線是另一個次要選項。 但是,在體驗了我的模型之后,我們可以通過為 EDGE_TPU 提供多個單一輸入作為另一種選擇來制作偽批處理

在此處輸入圖片說明

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM