add dynamic export and visualize to pytorch2onnx (#463)
* add dynamic export and visualize to pytorch2onnx * update document * fix lint * fix dynamic error and add visualization * fix lint * update docstring * update doc * Update help info for --show Co-authored-by:Jerry Jiarui XU <xvjiarui0826@gmail.com> * fix lint Co-authored-by:
maningsheng <maningsheng@sensetime.com> Co-authored-by:
Jerry Jiarui XU <xvjiarui0826@gmail.com>
Showing
- docs/useful_tools.md 24 additions, 2 deletionsdocs/useful_tools.md
- mmseg/apis/inference.py 10 additions, 2 deletionsmmseg/apis/inference.py
- mmseg/models/segmentors/encoder_decoder.py 6 additions, 1 deletionmmseg/models/segmentors/encoder_decoder.py
- mmseg/ops/wrappers.py 0 additions, 3 deletionsmmseg/ops/wrappers.py
- tools/pytorch2onnx.py 162 additions, 18 deletionstools/pytorch2onnx.py
Loading
Please register or sign in to comment