-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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 onnx2tensorrt #317
Add onnx2tensorrt #317
Conversation
Codecov Report
@@ Coverage Diff @@
## master #317 +/- ##
=======================================
Coverage 80.92% 80.92%
=======================================
Files 168 168
Lines 8409 8409
Branches 1216 1216
=======================================
Hits 6805 6805
Misses 1454 1454
Partials 150 150
Flags with carried forward coverage won't be shown. Click here to find out more. Continue to review full report at Codecov.
|
@grimoire Could you review this PR? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
@innerlee See if you have any comments on this PR. |
Yeah tomorrow |
* Adding onnx2tensorrt * Adding onnx2tensorrt * add args and trt version * add args and trt version * reuse function * add link of 40th line * init * Delete et HEAD~3
Hi
This PR adds tools to convert the ONNX model to TensorRT.
ESRGAN and SRCNN have been tested.
Note that SRCNN only support 'bilinear' upsampling.