Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

无法导出onnx以及tensorrt #1029

Open
3 tasks done
TtopsBo opened this issue Sep 2, 2024 · 0 comments
Open
3 tasks done

无法导出onnx以及tensorrt #1029

TtopsBo opened this issue Sep 2, 2024 · 0 comments

Comments

@TtopsBo
Copy link

TtopsBo commented Sep 2, 2024

Prerequisite

🐞 Describe the bug

使用15分钟Object detection教程中的export tensorrt 无法得到engine文件
(openmmlab) D:\mmyolo>python ./projects/easydeploy/tools/export_onnx.py configs/rtmdet/config.py D:/RTMdet/weights/epoch_25.pth --work-dir work_dir --img-size 640 640 --batch 1 --device cuda:0 --simplify --opset 11 --backend TENSORRT8 --pre-topk 1000 --keep-topk 100 --iou-threshold 0.65 --score-threshold 0.25
Export ONNX with bbox decoder and NMS ...
Loads checkpoint by local backend from path: D:/RTMdet/weights/epoch_25.pth
运行之后检查work_dir为空 是我使用的customized的config没有配置好的原因吗

Environment

cudatoolkit 10.2.89 he40447d_13
cudnn 8.4.1.50 hf5f08ae_0
onnx 1.16.2 pypi_0 pypi
onnxconverter-common 1.14.0 pypi_0 pypi
onnxruntime 1.8.1 pypi_0 pypi
onnxruntime-gpu 1.14.1 pypi_0 pypi
opencv-python 4.10.0.84 pypi_0 pypi
torch 1.10.1+cu102 pypi_0 pypi
torchaudio 0.10.1+cu102 pypi_0 pypi
torchvision 0.11.2+cu102 pypi_0 pypi

Additional information

No response

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant