27a856637c
fix mmcls ut
2022-07-11 19:10:04 +08:00
41a97746cf
finish mmcls ut
2022-06-22 22:32:14 +08:00
843d3c9519
Rewrite mmcls gap ( #478 )
...
* rewrite mmcls gap to replace view with flatten
* add test
* update
2022-05-17 19:57:38 +08:00
2c2d1e5ad9
feat(codebase/cls): support vision_transformer ( #403 )
...
* feat(codebase/cls): support vision_transformer
* style(onnx2ncnn): format cpp code, upgrade mmcls version
* fix(CI): upgrade mmcv to 1.4.2
* fix(onnx2ncnn): offset out of range during fuse conv reshape
* docs(vision_transformer.py): update VisionTransformer desc
* docs(onnx2ncnn.cpp): add more comment
* feat(onnx2ncnn.cpp): revert fuse weight
* docs(onnx2ncnn.cpp): add more comment
* test(vision_transformer): add test case
* refactor(vision_transformer.py): use symbol rewrite layer_norm
* refactor(vision_transformer): fix review
* fix(attention): add missing files
2022-04-26 18:00:38 +08:00
787ebc2392
[Feature]: Support mmpose ( #94 )
...
* add mmpose code
* update
* update
* add rewrites
* test trt
* test litehrnet with trt
* revert unused change
* add docs about mmpose
* add docstring and staticmethod
* update
* update
* update docs
* fix config name and docs
* add pose_detection ut
* add pose data
* fix lint of model.py
* add pose_detection_model ut
* fix docs and docstrinf
* add test_mmpose_models.py
* fix yapf
* fix lint
* fix create input
* support ort ut
* fix yapf
* fix docs
* fix createinput
* test ci bug
* rm test1.py
* fix yapf
* fix flake8
* fix yapf
* add config and update benchmark
* fix table format
* update mmpose benchmark
* update benchmark for mmpose
* run mmpose tests seperately in ci
* fix lint
* resolve comments
* add trt ut config
* fix test
* fix tests
* resolve comments
* resolve comments
* update tests
Co-authored-by: VVsssssk <shenkun@pjlab.org.cn>
Co-authored-by: hanrui1sensetime <hanrui1@sensetime.com>
2022-02-16 11:03:12 +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
4149228716
[Enhancement]: Import codebase only when it is required ( #266 )
...
* Add import codebase
* lint
* Fix import order
* typo
* Fix partition
* docstring
* lint
2021-12-10 11:34:22 +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
cab362a248
[Unittest] mmcls fix ut ( #228 )
...
* mmcls fix ut
* force add data folder
* remove debug info
* fix ut bugs
2021-11-25 18:48:11 +08:00
grimoire
RunningLeon
tpoisonooo
RunningLeon
Yifan Zhou
q.yao
hanrui1sensetime