kotlindl icon indicating copy to clipboard operation
kotlindl copied to clipboard

Add validation of input shape during onnx model inference

Open ermolenkodev opened this issue 2 years ago • 0 comments

Currently, if an onnx model requires a 3-channel image but the input provided has only one channel internal onnx library exception will be thrown.

ai.onnxruntime.OrtException: Shape [1, 512, 512, 3], requires 786432 elements but the buffer has 262144 elements.

ermolenkodev avatar May 26 '22 11:05 ermolenkodev