Closed ukicomputers closed 10 months ago
use the provided convertPytorchToONNX.py
script for the conversion as I adopted the same output format as the other YOLO versions, You can refer to the following log :
[INFO ] Name [0]: outputs
[INFO ] Shape [0]: (1, 8400, 85, )
[INFO ] Class num: 80
Hi! I have an converted (not custom, default COCO) Yolo-NAS-S model from default .pth to ONNX. When I execute the program by using following arguments:
./demo -m ~/models/dnn/yolo_nas_s.onnx -s 0 -c ~/Downloads/coco.names
I get this error when trying to execute on video detection (this is whole log):
And this error with image, even if given image is properly working, ends .jpg as code needs, and it is tested with OpenCV test code, what gives proper results. This is also an full log:
I am using the latest version of ONNX Runtime,
1.16.3
. The ONNX file is tested, it works properly.So when I debugged with Visual Studio Code, I get the error specifically at this line, when it tries to get an outputNames.size(), e.g. when it runs an session.Run (get detection results):
Do you know what maybe is issue? Thanks in advance. If you need more info, please leave a comment.