search

opencv / open_vision_capsules

A set of libraries for encapsulating smart vision algorithms
Other
130 stars 48 forks source link

Add a tensorrt backend #33

Open BestDriverCN opened 3 years ago

BestDriverCN commented 3 years ago

Add BaseTensorRTBackend that provide support for TensorRT models

apockill commented 3 years ago

I'll add the rest of my comments once there's a clear story for mutli-GPU support.

BestDriverCN commented 3 years ago

@apockill I resolved most of your comments but the post-process stuff. The code is tested to be working and the performance is almost the same. I'll continue to work on the GridParser stuff. In the meantime, you can take another look.