ctuning / ck-tensorflow

Collective Knowledge components for TensorFlow (code, data sets, models, packages, workflows):
http://cKnowledge.org
BSD 3-Clause "New" or "Revised" License
93 stars 26 forks source link

Reworked object-detection-tf-py to support batch, custom models and tensorRT #131

Closed me2x closed 5 years ago

me2x commented 5 years ago

major rework on the detect.py module, to encapsulate the main functions. This allows to create hooks in the main loop, where those functions can be substituted with custom implementations to support different models or different inference engines like tensorRT.

I also provided those function for tensorRT.

Moreover, there are a couple of modifications in the installation of TF-cuda, where I inserted the possibility to select tensorRT and I had to modify the install script to insert a boundary on the register count when compiling from source, which was otherwise failing.