将 YOLOv5 ONNX 模型转换为中间表示 (IR) 格式时,收到与节点相关的错误
内容类型: 错误讯息 | 文章 ID: 000088394 | 上次审核日期: 2022 年 05 月 20 日
python /opt/intel/openvino_2021/deployment_tools/model_optimizer/mo.py --input_model "model.onnx" --output_dir "cur_folder/"
[ ERROR ] The ExpandDims node Unsqueeze_583 has more than 1 input
[ ERROR ] Cannot infer shapes or values for node "Slice_4".
[ ERROR ] Output shape: [0 3 0 0] of node "Slice_4" contains non-positive values
[ ERROR ] It can happen due to bug in custom shape infer function .
[ ERROR ] Or because the node inputs have incorrect values/shapes.
[ ERROR ] Or because input shapes are incorrect (embedded to the model or passed via --input_shape).
YOLOv5 中有三个输出节点,所有节点都需要在命令中指定:
模型优化器命令: python mo.py --input_model model.onnx --output --input_shape [1,3,512,512]
使用 Netron 可视化 YOLOv5 ONNX 重量。