-
目前仅将onnx导出,然后后处理仍然使用的detectron中的函数
- 进阶:将RoiAlign作为插件集成到onnx/trt中
-
使用流程
-
先安装环境
torch 1.10.2+cu102 torchvision 0.11.3+cu102 vs2019
- 安装detectron
- windows
- linux
- 安装官方命令即可。
- 安装detectron
-
-
导出onnx
运行 mask_demo.py 导出onnx。
-
生成engine
trtexec --onnx=./mask.onnx --saveEngine=./mask_fp32.engine --workspace=1000
-
运行
执行trt_py.py脚本