Skip to content

Conversation

@stefanopini
Copy link
Owner

@stefanopini stefanopini commented Nov 21, 2022

Add TensorRT Support from #14 (in branch tensorrt) to master branch.

  • Clean up code
  • Update docstrings
  • Add missing params
  • Test on Colab
  • Add documentation
  • Add colab link

* [Add]: TensorRT Support

* [Fix]: Script improvement

* [Fix]: Script improvement

* [Fix]: Address the requested changes

Co-authored-by: Gpastal24 <gpastal24@users.noreply.github.com>
@stefanopini
Copy link
Owner Author

Hi @gpastal24 , could you please take a look?
It looks good to me, I also tested it on Google Colab (see also added notebook) and it worked with no issues at all!

@gpastal24
Copy link
Contributor

Hi @gpastal24 , could you please take a look? It looks good to me, I also tested it on Google Colab (see also added notebook) and it worked with no issues at all!

Hello @stefanopini I tested the colab notebook in the morning , I think it works ok and the detections seemed fine although I still got the warning that trt was linked at a different cudnn. Nonetheless and if someone experiences issues he can always downgrade the trt version.
It is good to go from my side

@stefanopini
Copy link
Owner Author

Thanks a lot for checking and testing the whole notebook! I'll merge into master

@stefanopini stefanopini merged commit fee3011 into master Nov 23, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants