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 onnx2tensorrt #317

Merged
merged 10 commits into from
May 27, 2021
Merged

Conversation

AllentDan
Copy link
Member

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.

@innerlee innerlee requested a review from RunningLeon May 18, 2021 08:36
@innerlee innerlee changed the title Adding onnx2tensorrt Add onnx2tensorrt May 18, 2021
@codecov
Copy link

codecov bot commented May 18, 2021

Codecov Report

Merging #317 (fe190dd) into master (e6406c1) will not change coverage.
The diff coverage is n/a.

❗ Current head fe190dd differs from pull request most recent head cfed165. Consider uploading reports for the commit cfed165 to get more accurate results
Impacted file tree graph

@@           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           
Flag Coverage Δ
unittests 80.92% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.


Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update e6406c1...cfed165. Read the comment docs.

docs/tools_scripts.md Outdated Show resolved Hide resolved
@RunningLeon
Copy link
Collaborator

@grimoire Could you review this PR?

docs/tools_scripts.md Show resolved Hide resolved
tools/onnx2tensorrt.py Outdated Show resolved Hide resolved
tools/onnx2tensorrt.py Outdated Show resolved Hide resolved
Copy link
Member

@grimoire grimoire left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

docs/tools_scripts.md Outdated Show resolved Hide resolved
tools/onnx2tensorrt.py Outdated Show resolved Hide resolved
et HEAD~3 Outdated Show resolved Hide resolved
@RunningLeon RunningLeon requested a review from innerlee May 25, 2021 04:31
@RunningLeon
Copy link
Collaborator

@innerlee See if you have any comments on this PR.

@innerlee
Copy link
Contributor

Yeah tomorrow

@innerlee innerlee merged commit 98a7ca3 into open-mmlab:master May 27, 2021
Yshuo-Li pushed a commit to Yshuo-Li/mmediting that referenced this pull request Jul 15, 2022
* 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
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

4 participants