python - OpenVino 模型在加载网络后崩溃
问题描述
我使用将 onnx 模型转换为 IR
python3 mo_onnx.py --input_model SSD_nms.onnx --input inputs,top_k_per_class,nms_iou,nms_score --input_shape [1,3,224,224],[1],[1],[1]
这一步工作正常,但是当我尝试加载网络时,它给了我以下错误。
ie = IECore()
net = ie.read_network(model=model_path, weights=weights_path)
print (net.layers)
File "ie_api.pyx", line 1170, in openvino.inference_engine.ie_api.IENetwork.layers.__get__ ValueError: get_shape was called on a descriptor::Tensor with dynamic shape
我能够成功运行相同的模型减去最后一个 NMS 层。不知道为什么这会失败。在 netron 上查看 IR xml 文件并没有显示任何内容。
导致麻烦的整个层描述:
<layer id="351" name="selected_indices" type="NonMaxSuppression" version="opset1">
<data box_encoding="center" sort_result_descending="0"/>
<input>
<port id="0">
<dim>1</dim>
<dim>10000</dim>
<dim>4</dim>
</port>
<port id="1">
<dim>1</dim>
<dim>2</dim>
<dim>10000</dim>
</port>
<port id="2"/>
<port id="3"/>
<port id="4"/>
</input>
<output>
<port id="5" precision="I64">
<dim>10000</dim>
<dim>3</dim>
</port>
</output>
</layer>
从 IR 中移除输出层会使事情正常工作。
解决方案
推荐阅读
- powerbi - 将默认下拉值更改为当前年份
- c++ - 为什么 C++ 不喜欢这种字符串语法?
- c# - 使用网络表单和 EPPlus 下载文件
- wix - WiX 有条件地更改产品名称
- custom-element - reportValidity() 与表单相关的 customElement 失败
- validation - 如何根据单元格值显示隐藏行?
- json - 在 Groovy 中将 Json 字符串解析为 GStrings
- php - Jquery onchange 后动态选择框未正确填充 TypeAhead 输入文本
- bayesian - 为什么即使我不包含任何观察结果 pymc3 也会运行?
- javascript - Google Scheet 和 Survey Monkey API 插件上的时间触发器