ibaiGorordo / ONNX-CREStereo-Depth-Estimation

Python scripts performing stereo depth estimation using the CREStereo model in ONNX.
MIT License
125 stars 27 forks source link

[ONNXRuntimeError] : 1 : FAIL : Non-zero status code returned while running Mul node. #5

Open kxzhang-lab opened 1 year ago

kxzhang-lab commented 1 year ago

When I run the example"python driving_stereo_point_cloud.py",I occur an error as the following: 2022-11-29 10:27:24.272331245 [E:onnxruntime:, sequential_executor.cc:333 Execute] Non-zero status code returned while running Mul node. Name:'init_Mul_498' Status Message: CUDA error cudaErrorInvalidConfiguration:invalid configuration argument Traceback (most recent call last): File "driving_stereo_point_cloud.py", line 65, in depth_estimator(left_img, right_img) File "/media/fhy/TOSHIBA_EXT/ZKX/ONNX_CREStereo_Depth_Estimation/ONNX-CREStereo-Depth-Estimation-main/crestereo/crestereo.py", line 23, in call return self.update(left_img, right_img) File "/media/fhy/TOSHIBA_EXT/ZKX/ONNX_CREStereo_Depth_Estimation/ONNX-CREStereo-Depth-Estimation-main/crestereo/crestereo.py", line 57, in update left_tensor, right_tensor) File "/media/fhy/TOSHIBA_EXT/ZKX/ONNX_CREStereo_Depth_Estimation/ONNX-CREStereo-Depth-Estimation-main/crestereo/crestereo.py", line 97, in inference_with_flow self.input_names[3]: right_tensor})[0] File "/home/fhy/anaconda3/envs/CREStereo/lib/python3.7/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 124, 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 Mul node. Name:'init_Mul_498' Status Message: CUDA error cudaErrorInvalidConfiguration:invalid configuration argument.

when I run the example"python image_depth_estimation.py",I occur an error as the following: 2022-11-29 10:26:40.240696007 [E:onnxruntime:, sequential_executor.cc:333 Execute] Non-zero status code returned while running Mul node. Name:'init_Mul_498' Status Message: CUDA error cudaErrorInvalidConfiguration:invalid configuration argument Traceback (most recent call last): File "image_depth_estimation.py", line 28, in disparity_map = depth_estimator(left_img, right_img) File "/media/fhy/TOSHIBA_EXT/ZKX/ONNX_CREStereo_Depth_Estimation/ONNX-CREStereo-Depth-Estimation-main/crestereo/crestereo.py", line 23, in call return self.update(left_img, right_img) File "/media/fhy/TOSHIBA_EXT/ZKX/ONNX_CREStereo_Depth_Estimation/ONNX-CREStereo-Depth-Estimation-main/crestereo/crestereo.py", line 57, in update left_tensor, right_tensor) File "/media/fhy/TOSHIBA_EXT/ZKX/ONNX_CREStereo_Depth_Estimation/ONNX-CREStereo-Depth-Estimation-main/crestereo/crestereo.py", line 97, in inference_with_flow self.input_names[3]: right_tensor})[0] File "/home/fhy/anaconda3/envs/CREStereo/lib/python3.7/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 124, 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 Mul node. Name:'init_Mul_498' Status Message: CUDA error cudaErrorInvalidConfiguration:invalid configuration argument