#在使用openvino推理训练出来的模型时,这时候需要将最先训练好的pt权重转化为onnx模型后再转换为ir模型#
[ ERROR ] Cannot infer shapes or values for node "/model.24/Reshape".
[ ERROR ] Number of elements in input [ 1 255 80 80] and output [1, 3, 85, 80, 60] of reshape node /model.24/Reshape mismatch
[ ERROR ]
[ ERROR ] It can happen due to bug in custom shape infer function <function Reshape.infer at 0x000001ECEC43C0D0>.
[ 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).
[ ERROR ] Run Model Optimizer with --log_level=DEBUG for more information.
[ ERROR ] Exception occurred during running replacer "REPLACEMENT_ID" (<class 'extensions.middle.PartialInfer.PartialInfer'>): Stopped shape/value propagation at "/model.24/Reshape" node.
For more information please refer to Model Optimizer FAQ, question #38. (https://docs.openvinotoolkit.org/latest/openvino_docs_MO_DG_prepare_model_Model_Optimizer_FAQ.html?question=38#question-38)
上图报错是在进行onnx转ir模型时的报错,因为这个转ir模型需要进行两步转化,所以两步转换设置的参数应一致相同。
图中的报错就是我自己在使用yolov5的expert.py文件将pt文件转化为onnx时设置的输入大小时640*480,而在转ir形式时设置的时640*640,需要保证这两个值相同
所以出现这种错误可以查看两次转换的参数是否相同,或者是缺少了有些参数,因为有的代码,在没设置参数时会直接使用默认值,这样也会导致两次的参数不相同,导致报错。
如果友友们觉得有用的话吗,就点个赞吧,