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
- 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+
- After train your own model, go to
depolyand run it on tensorRT!