mmsegmentation/mmseg/models
q.yao 789d1a142b
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
..
backbones Save base_channels for UNet (#450) 2021-03-30 09:52:08 -07:00
decode_heads Fix typo: upsampe_cfg -> upsample_cfg (#449) 2021-03-30 09:51:38 -07:00
losses [Improvement] Add more CI (#460) 2021-04-08 19:29:00 -07:00
necks Add Semantic FPN (#94) 2020-09-03 19:56:36 +08:00
segmentors add dynamic export and visualize to pytorch2onnx (#463) 2021-04-12 11:54:59 -07:00
utils refactor test organization (#440) 2021-03-30 17:55:09 -07:00
__init__.py Add Semantic FPN (#94) 2020-09-03 19:56:36 +08:00
builder.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00