-
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
[Feature] Update pytorch2onnx #265
Conversation
Codecov Report
@@ Coverage Diff @@
## master #265 +/- ##
==========================================
- Coverage 81.34% 79.93% -1.41%
==========================================
Files 158 159 +1
Lines 7773 7941 +168
Branches 1152 1177 +25
==========================================
+ Hits 6323 6348 +25
- Misses 1306 1449 +143
Partials 144 144
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
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
* add restorer model onnx support, add visualize * use cv2 instead of matplotlib * fix dynamic batch output axis
Hi,
This PR adds restorer model support to pytorch2onnx.
Dynamic shape, dynamic batch and visualize support are also added too.