Open LeiGan0215 opened 3 years ago
作者您好,我按照您readme里面的步骤操作完成之后,示例图片的检测结果错误的,我用nanodet里面的python ./tools/demo.py检测结果是对的,应该是转onnx还是tensorrt的哪个步骤错误了
`import onnxruntime import numpy as np
sess_options = onnxruntime.SessionOptions() sess = onnxruntime.InferenceSession('././nanodet-m.onnx', sess_options) data = [np.random.rand(1, 3, 320, 320).astype(np.float32)] input_names = sess.getinputs() feed = zip(sorted(i.name for i_ in input_names), data) result = sess.run(None, dict(feed)) print(result[0].shape)`
参考这个方法测试onnx模型,输出是(1,2100,84),这个结果对吗