search

linghu8812 / tensorrt_inference

699 stars 205 forks source link

nandet检测结果错误 #80

Open LeiGan0215 opened 3 years ago

LeiGan0215 commented 3 years ago

捕获 作者您好，我按照您readme里面的步骤操作完成之后，示例图片的检测结果错误的，我用nanodet里面的python ./tools/demo.py检测结果是对的，应该是转onnx还是tensorrt的哪个步骤错误了

LeiGan0215 commented 3 years ago

`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），这个结果对吗