Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add TensorRT Support to master branch (from #14) #15

Merged
merged 11 commits into from
Nov 23, 2022
Merged

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.

None yet

2 participants