ibaiGorordo / ONNX-YOLOv8-Object-Detection

Python scripts performing object detection using the YOLOv8 model in ONNX.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

image_object_detection.py run error

osenberg-x opened this issue · comments

commented

└─[$]> python3 image_object_detection.py
Traceback (most recent call last):
File "/home/x/ws/ml/ONNX-YOLOv8-Object-Detection/image_object_detection.py", line 16, in
boxes, scores, class_ids = yolov8_detector(img)
File "/home/x/ws/ml/ONNX-YOLOv8-Object-Detection/yolov8/YOLOv8.py", line 19, in call
return self.detect_objects(image)
File "/home/x/ws/ml/ONNX-YOLOv8-Object-Detection/yolov8/YOLOv8.py", line 31, in detect_objects
input_tensor = self.prepare_input(image)
File "/home/x/ws/ml/ONNX-YOLOv8-Object-Detection/yolov8/YOLOv8.py", line 46, in prepare_input
input_img = cv2.resize(input_img, (self.input_width, self.input_height))
cv2.error: OpenCV(4.8.0) 👎 error: (-5:Bad argument) in function 'resize'

Overload resolution failed:

  • Can't parse 'dsize'. Sequence item with index 0 has a wrong type
  • Can't parse 'dsize'. Sequence item with index 0 has a wrong type

What is the input size of your model? Are you using an ONNX model with dynamic input size? (That probably won't work)

@ibaiGorordo u have names missmatch and wrong resize function in YOLOv8.py:
input_img = cv2.resize(input_img, (int(self.img_width), int(self.img_height)),interpolation=cv2.INTER_AREA)

but even with fix it doesnt work:

File "/home/pi/.local/lib/python3.9/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 200, in run
    return self._sess.run(output_names, input_feed, run_options)
onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Non-zero status code returned while running Concat node. Name:'/model.11/Concat' Status Message: concat.cc:159 PrepareForCompute Non concat axis dimensions must match: Axis 2 has mismatched dimensions of 45 and 46