Add onnx convert script (#2291)
* add onnx convert script * upload pytorch2onnx.py * restore config * set use_torchvion on-the-fly, update doc * update doc * support pass input * fixed passes
Showing
- .isort.cfg 1 addition, 1 deletion.isort.cfg
- docs/GETTING_STARTED.md 9 additions, 0 deletionsdocs/GETTING_STARTED.md
- mmdet/models/detectors/cascade_rcnn.py 1 addition, 1 deletionmmdet/models/detectors/cascade_rcnn.py
- mmdet/models/detectors/double_head_rcnn.py 1 addition, 1 deletionmmdet/models/detectors/double_head_rcnn.py
- mmdet/models/detectors/grid_rcnn.py 1 addition, 1 deletionmmdet/models/detectors/grid_rcnn.py
- mmdet/models/detectors/htc.py 1 addition, 1 deletionmmdet/models/detectors/htc.py
- mmdet/models/detectors/two_stage.py 1 addition, 1 deletionmmdet/models/detectors/two_stage.py
- tools/pytorch2onnx.py 125 additions, 0 deletionstools/pytorch2onnx.py
Loading
Please register or sign in to comment