9eee75b98f
More ocr models ( #446 )
...
* rewrite sync batchnorm
* export panet and psenet
* resolution
* align fp16 for panet
* refine codes
* enable satrn for trt
* refine docs
* docstring
* docstring
* add ut and refine codes
* fix ut
* resolve comments and move panet-fp16 to doc
* remove ut
* refine ut
* resolve comments
* use size instead of img_scale
* use size of MultiScaleAug
Co-authored-by: dongchunyu.vendor <dongchunyu@pjlab.org.cn>
2022-05-20 21:52:36 +08:00
aa85760531
[Enhancement] Support litehrnet for ncnn ( #316 )
...
* support for litehrnet
* support ncnn vulkan
* update supported models
* add docstring
* add one ut and fix wrapper
* add test_shuffleunit
* add test_stem_forward
* add last ut
* fix flake8
* fix yapf
* fix lint
* fix yapf
* fix comments
* unified chunk
* mv adaptive_avg_pool
* fix lint
* move adaptive_avg_pool_ncnn
* fix lint
* symbolic rewriter of adaptive_avg_pool2d
* fix lint
* fix yapf
* fix ci
2022-04-29 16:46:19 +08:00
33bde90744
[Fix] Fix unittest of ncnn. ( #309 )
...
* fix test_pytorch_functions
* fix test_mmocr_models
2021-12-21 18:03:50 +08:00
fabdb473bb
Update ncnn test ( #298 )
...
* update ncnn test
* type hint
* update test ocr
* update mmseg ut
* ignore ncnn rpn head test
* add logging
* fix ssd base dense head test
* recover bacth in ncnn wrapper
* fix ncnn_ops_ut
* fix yapf
* recover test ops
* fix run_with_backend False
* Revert "fix run_with_backend False"
This reverts commit 83f8f915a25e800f5c2db339584d164ba40b2d9b.
* disable ncnn test test_pytorch_functions.py
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: hanrui1sensetime <hanrui1@sensetime.com>
2021-12-20 14:05:13 +08:00
0b23c744fc
remove trt in config ( #280 )
2021-12-13 19:57:15 +08:00
bb9b0a98de
[Enhancement]: Added static config and CI tests for OpenVINO. ( #218 )
...
* Add openvino_static.
* Add openvino-dev.
* Fix skipping ORT tests in test_mmocr_models.
* Updated docs.
* Fix print.
* Fix
* Fix
* Fix other backends
* Fix is_available
* fix ncnn
* Add constrict for get rewrite output
* add not
* Fix
* fix
* Fix
* Fix
* Improve tests
* Remove rebundant `cuda`
* Prevent None object and rename variable
* Fix multi-line string
* rename get_backend_checker
* Add Troubleshooting to doc.
* Fix postprocessing_masks with empty masks.
* Fix tests
* lint
* Update docs.
Co-authored-by: SingleZombie <singlezombie@163.com>
2021-12-01 14:03:48 +08:00
4fc0e79b4e
[Enhancement]: Refine backend environment check and fix online unit test ( #239 )
...
* Fix
* Fix other backends
* Fix is_available
* fix ncnn
* Add constrict for get rewrite output
* add not
* Fix
* fix
* Fix
* Fix
* Improve tests
* Remove rebundant `cuda`
* Prevent None object and rename variable
* Fix multi-line string
* rename get_backend_checker
* Fix tests
* lint
* docstring
2021-12-01 10:19:11 +08:00
a8abc75065
Add licence head to code ( #236 )
2021-11-30 15:00:37 +08:00
eab6d76838
Fix test ( #234 )
2021-11-25 19:41:31 +08:00
6f36717a72
[Unittest]: Test mmdeploy.pytorch ( #133 )
...
* add some tests
* Fix topk bug
* Fix bug in test
* Add remaining tests
* Remvoe debug code
* Delete redundant code
* Remove redundant code
2021-10-19 13:14:33 +08:00
AllentDan
hanrui1sensetime
RunningLeon
Semyon Bevzyuk
Yifan Zhou
q.yao