onnx模型轉openvino報錯nGraph does not support the following ONNX operations: If

hyliuisme發表於2020-11-08

onnx模型轉換openvino進行部署時首先需要用openvino官方給的優化程式對onnx模型進行優化,之後使用openvino官方API載入優化後的模型如果不報錯說明onnx模型轉openvino模型成功,就可以編寫推理程式碼然後進行部署。我在使用優化程式轉換onnx模型時雖無報錯但是在使用openvino官方API載入模型時報錯nGraph does not support the following ONNX operations: If,這個錯誤好像是onnx模型轉openvino模型不支援if條件語句,因此我首先嚐試將程式碼裡所有if語句去掉,但還是報這個錯誤,在經過仔細排查之後發現pytorch的squeeze()方法方法在轉換成onnx模型後會產生If節點,最終導致轉換openvino模型出現nGraph does not support the following ONNX operations: If這個報錯,squeeze()方法只是去掉tensor中維度為1的維度,並不清楚為何在轉換成onnx模型後會產生If節點。

相關文章