Junjun2016
2fe0bddf5e
[Dcos] Add header for files ( #796 )
...
* Add header for files
* Delete header in config files
2021-08-16 23:16:55 -07:00
Miguel Méndez
50461efe85
[Fix] Replace interpolate with resize ( #731 )
...
* Replace interpolate with resize
* Replace nn.Upsample with ops.Upsample
* Fix test
2021-07-28 16:56:22 +08:00
q.yao
b49e86a5b8
fix verify bugs ( #547 )
...
* fix verify bugs
* rename args
2021-05-12 21:14:41 -07:00
q.yao
6ccb1c0fe5
[Feature] add onnxruntime test tool ( #498 )
...
* add onnxruntime test tool, update pytorch2onnx to support slice export
* onnx convert with custom output shape, update test code
* update pytorch2onnx, add rescale_shape support, add document
* update doc for lint error fixing
* remove cpu flag in ort_test.py
* change class name, fix cuda error
* remote comment
* fix bug of torch2onnx
* mIOU to mIoU
2021-04-28 20:38:01 -07:00
q.yao
bc2dc1277a
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>
2021-04-12 11:54:59 -07:00
Jerry Jiarui XU
4423d32702
[Improvement] Move train_cfg/test_cfg inside model ( #341 )
...
* [Improvement] Move train_cfg/test_cfg inside model
* fixed config dict
* fixed config dict
2021-01-19 17:06:23 -08:00
Lei Yang
93a2456def
Remove redundancies in pytorch2onnx ( #160 )
...
* rm redundancies
* re-add some packages
2020-09-30 10:11:34 +08:00
robin Han
2e5260b58a
Onnx upsample ( #100 )
...
* add customized Upsample which can convert to ONNX
* support multiply decode head for hrnet
* support size for Upsample
2020-09-03 19:59:13 +08:00
robin Han
5ddef979fc
add pytorch2onnx part ( #12 )
...
* add pytorch2onnx part
* Update according to the latest mmcv
* add docstring
* update docs
* update docs
Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-08-14 03:28:21 +08:00