linghu8812
linghu8812
@DataXujing 请确认一下config.yaml文件中的ONNX文件路径对不对,写个绝对路径试试。
@DataXujing 请问一下用的是哪个版本的TensorRT?请试一下这个onnx:[https://pan.baidu.com/s/1Sp-sOT_mYYVXgXE9uN_ShQ](https://pan.baidu.com/s/1Sp-sOT_mYYVXgXE9uN_ShQ),提取码:hytj
@DataXujing https://github.com/linghu8812/tensorrt_inference/blob/bdb3ac319e634f162984a32c93636ab50ae55a2f/ScaledYOLOv4/export_onnx.py#L55 导出ONNX模型时,试一试opset=10,7,0.0.11不支持opset=12
导出ONNX模型时,试一试opset=10,7,0.0.11及以下不支持opset=12
@sporterman 试一下下面代码测试yolov5s.onnx是否正确 ``` import onnxruntime import numpy as np sess_options = onnxruntime.SessionOptions() sess = onnxruntime.InferenceSession('./yolov5s.onnx', sess_options) data = [np.random.rand(1, 3, 640, 640).astype(np.float32)] input_names = sess.get_inputs() feed = zip(sorted(i_.name for i_...
> @sporterman 试一下下面代码测试yolov5s.onnx是否正确 > > ``` > import onnxruntime > import numpy as np > > sess_options = onnxruntime.SessionOptions() > sess = onnxruntime.InferenceSession('./yolov5s.onnx', sess_options) > data = [np.random.rand(1, 3, 640,...
@IGnoredBird 与pytorch不同,修改mxnet导出的onnx模型opset,需要修改mxnet contrib内部的代码,试试使用onnx==1.5.0
@deep-practice 7.0.0.11 use opset=10 when export onnx models
@DaChaoXc TensorRT 7.1以上支持opset12,6.0和7.0支持opset10,高版本的onnx可以支持低opset
> Hello, > > I am using the default pretrained yolov7.pt model and followed steps on this repo to export into onnx file (using your export fork) and then attempt...