Traceback (most recent call last): File "C:\Users\12525\Desktop\python-raspberry-pi-master\EXP-yolov5-lite\my_test_one_img.py", line 148, in <module> det_boxes,scores,ids = infer_img(img0,net,model_h,model_w,nl,na,stride,anchor_grid,thred_nms=0.4,thred_cond=0.5) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "C:\Users\12525\Desktop\python-raspberry-pi-master\EXP-yolov5-lite\my_test_one_img.py", line 95, in infer_img outs = net.run(None, {net.get_inputs()[0].name: blob})[0].squeeze(axis=0) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "D:\app_python3.11\Lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 217, in run return self._sess.run(output_names, input_feed, run_options) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: images for the following indices index: 0 Got: 1 Expected: 8 Please fix either the inputs or the model.
时间: 2023-08-06 15:03:14 浏览: 223
alipay-sdk-python-all:支付宝开放平台 Alipay SDK for Python
根据错误信息,你的代码中出现了与之前相同的错误。这个错误是由于输入的维度不正确导致的。模型期望输入张量的第一个维度为8,但实际上你提供的输入张量的第一个维度为1。
你需要检查你的代码中如何生成和传递输入张量。确保输入张量的维度与模型的要求相匹配。
如果你使用的是预训练模型,你可以查看模型的文档或示例代码,了解输入张量的正确形状和维度。
如果你自己训练了模型,你需要检查训练数据的维度是否正确,并且在预测时按照相同的方式处理输入数据。
如果你需要更具体的帮助,请提供更多关于你使用的模型和代码的信息。
阅读全文