Open osenberg-x opened 1 year ago
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
└─[$]> 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) :-1: error: (-5:Bad argument) in function 'resize'