9849a8dc23
[Refactor]: Unified parameter initialization ( #567 )
...
* [Refactor]: Unified parameter initialization
* fixed pretrained
2021-06-16 21:41:29 -07:00
02b5d768aa
[feature]: Able to use save_best option ( #575 )
...
* Add save_best option in eval_hook.
* Update meta to fix best model can not test bug
* refactor with _do_evaluate
* remove redundent
* add meta
Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2021-06-02 16:25:26 -07:00
597736288c
[Feature] Update deploy test tools ( #553 )
...
* add trt test tool
* create deploy_test, update document
* fix with isort
* move import inside __init__
* remove comment, fix doc
* update document
2021-05-24 20:37:46 -07:00
b49e86a5b8
fix verify bugs ( #547 )
...
* fix verify bugs
* rename args
2021-05-12 21:14:41 -07:00
5182fa1523
add onnx to tensorrt tools ( #542 )
2021-05-11 20:02:27 -07:00
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
e5007e7491
[Bug fix] fixed fp16 inference ( #497 )
...
* fixed fp16
* update fps
2021-04-24 09:57:33 -07:00
c3d9642e2d
add tool pytorch2torchscript ( #469 )
...
* add tool pytorch2torchscript
* fix the assert message for pytorch version.
2021-04-19 08:51:49 -07:00
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
fd486cdbc3
add print model graph args for tools/print_config.py ( #451 )
...
* add print model graph for print_config tool
* add print model graph for print_config tool
* fix double quoted
* fix iosort
2021-04-06 15:14:54 -07:00
9cbb4b1288
Add opacity option to show_result ( #425 )
2021-03-22 20:34:38 -07:00
1722010396
add plot_logs tool ( #426 )
...
* Support plot logs
* add plot log docs
2021-03-21 22:05:32 -07:00
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
7e156454a2
memory efficient test ( #330 )
...
* memory efficient test
* implement efficient test
* merge
* Add document and docstring
* fix unit test
* add memory usage report
2021-01-09 23:47:31 -08:00
aba6d62c65
Doc refactor ( #311 )
...
* refactor docs
* add docs
* add modelzoo
* refactor getting started
2020-12-22 18:36:49 -08:00
294a1f377a
fix dataset jpg bug ( #213 )
...
* fix dataset jpg bug
* fix syntax error
2020-10-28 11:58:44 -07:00
464ffb77ad
Define blood vessel dataset and fix filename bug ( #203 )
...
* fix filename bug
* define blood vessel dataset
* redo debug
* fix small bug
* rename dataset
Co-authored-by: yamengxi <yamengxi@sensetime.com>
2020-10-22 17:27:40 -07:00
eaefe54e8d
Add blood vessel dataset processing script ( #184 )
...
* Add blood vessel dataset processing script
* Fix syntax error
* Fix syntax error
* Fix syntax error
* Fix bugs
* Fix bugs
* Fix bugs
* Use safe functions and expand more apis
* Use safe functions and expand more apis
* Fix hard code and verify dataset integrity
2020-10-18 01:22:42 -07:00
93a2456def
Remove redundancies in pytorch2onnx ( #160 )
...
* rm redundancies
* re-add some packages
2020-09-30 10:11:34 +08:00
7baed6513e
Add Pascal Context to mmsegmentation ( #133 )
...
* Add Pascal Context to mmsegmentation
* Add benchmark result to Pascal Context
* fix mmcv version
* fix code syntax
* fix code syntax again
* Update mmseg/models/segmentors/encoder_decoder.py
update hint
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
* update comment
* fix pascal context model path
* fix model path mistake again
* fix model path mistake again
* fix model path mistakes again
Co-authored-by: Jerry Jiarui XU <xvjiarui0826@gmail.com>
2020-09-22 14:56:13 +08:00
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
8c0e093b31
Fixed slide inference ( #90 )
2020-08-25 20:01:01 +08:00
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
0f702f44af
Auto get version info and git hash ( #55 )
...
* Auto get version info and git hash
* bump 0.5.1 and update doc
* fixed docs
* Add change log
2020-08-11 19:23:35 +08:00
fd100e02c4
Add load-from flag ( #33 )
...
* Add load-from flag
* minor update
Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-08-06 21:17:49 +08:00
daa023836b
Fixed voc aug convert ( #19 )
...
* Fixed voc aug convert
* update getting_started.md
* add class balanced doc
2020-07-17 09:55:28 +08:00
2fc821ec13
Add test tutorial ( #9 )
...
* add test tutorial
* remote torch/torchvision from requirements
* update getting started
* rename drop_out_ratio -> dropout_ratio
2020-07-12 23:53:56 +08:00
b2724da80b
init commit
2020-07-10 02:39:01 +08:00
Jerry Jiarui XU
Yinhao Li
q.yao
q.yao
sshuair
David de la Iglesia Castro
谢昕辰
yamengxi
Lei Yang
robin Han
Evgeny Nizhibitsky