Commit Graph

11 Commits (ad35ec63562ad40e596928bdd955400864eb5d17)

Author SHA1 Message Date
Miao Zheng 4d00877b4c Add deprecation message for deploy tool ()
* Add deprecation message for deploy tool

* deploy test
2022-02-25 11:22:03 +08:00
MengzhangLI d4f5bb25fc [Enhancement] Upgrade isort in pre-commit hook () 2022-02-08 19:06:33 +08:00
Junjun2016 2fe0bddf5e [Dcos] Add header for files ()
* 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 ()
* 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 ()
* fix verify bugs

* rename args
2021-05-12 21:14:41 -07:00
q.yao 6ccb1c0fe5 [Feature] add onnxruntime test tool ()
* 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 ()
* 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 ()
* [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 ()
* rm redundancies

* re-add some packages
2020-09-30 10:11:34 +08:00
robin Han 2e5260b58a Onnx upsample ()
* 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 ()
* 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