簡體   English   中英

openvino model 的推理僅返回分數

[英]Inference on openvino model returns only scores

我的任務是使用 Intel Movidius 和 Raspberry Pi 執行人臉檢測推理。 錯誤是 model 僅返回“分數”-> (1, 3000, 2) 而不是“框”。

步驟:在我的本地機器上,我從存儲庫https://github.com/qfgaohao/pytorch-ssd訓練了幾個模型(mb1-ssd、mb1-ssd-lite、vgg16-ssd)並將它們轉換為 onnx。 然后,使用 openvinotoolkit = 2020.1 中的 open vino model 優化器,我獲得了每個 model 的“.bin”、“.xml”文件。 然后,使用獲得的文件,我在 Rasberry Pi 上執行了推斷並遇到了上述錯誤。

注意:推理工作使用來自 model 動物園的預訓練人臉檢測模型,我發現查看 .xml 文件和 my.xml 文件的唯一區別是缺少“檢測輸出”層。 但是,當我使用 netron 可視化 .xml 文件時,轉換似乎是正確的。

回購鏈接: https://github.com/cocacola0/bsc_thesis

OpenVINO™ 2020.3 版本是最后一個支持由英特爾® Movidius™ Myriad™ 2 提供支持的英特爾® Movidius™ 神經計算棒的 OpenVINO™ 版本。

使用ssd_mobilenet_v2_cocossdlite_mobilenet_v2作為 Open Model Zoo 提供的替代模型。 這兩種模型都適用於您的代碼。

暫無
暫無

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

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