zhangjinsong3 / DBNet.tensorRT

DBNet on onnx and tensorRT

DBNet.tensorRT:

note: code based on WenmuZhou/DBNet.pytorch

add support for TensorRT inference

• python
• C++
• full_dims and dynamic shape

git clone https://github.com/zhangjinsong3/DBNet.tensorRT.git
cd DBNet.tensorRT/deploy

Requirements

• pytorch 1.4+
• torchvision 0.5+
• gcc 4.9+
• tensorRT 6.0.1.5+
• onnx-tensorrt 6.0+
• onnx 1.4.0+
• opencv 3.4.2+

Deploy

• After train your own model, go to depoly and run it on tensorRT!