deaefacea1
Add softmax to cls model ( #1573 )
...
* Add softmax to cls model
* fix cls ci
* multihead
* update classification_model.py
2022-12-30 15:43:12 +08:00
6e91614171
CodeCamp #102 : support SOLO deployment with OpenVINO ( #1454 )
...
* feat: support solo mmdet3.x openvino 2.0 api
* feat: support solo mmdet3.x openvino
* fix: lint
* fix: add solo head test
* docs: add supported_modesl
* docs: add supported_models
* fix: fix unreasonable code
* fix: fix ci failed
* feat: add linspace func rewrite
* fix: fix unreasonable rewrite linspace__onnx
* fix: change func name from __onnx to __default
* feat: add solo test regression
2022-12-22 00:02:25 +08:00
40adb329a3
Merge master:ea7706cb into sync_master
2022-09-22 19:49:50 +08:00
b0b502cba0
[Fix] Sync codebase 2.0 ( #979 )
...
* remove mmcv.Config
* fix mmdet with new mmengine
* fix ut
* fix mmdet ut
* fix sdk cpp pipeline
* fix mmcls
* fix mmcls ut
* adapt for new config
* fix cls models
* fix mmcls sdk
* fix yapf
* fix lint
* fix torchscript
* fix torchscript cpu
* fix check_env
* add test data
* fix export_info
* fix blank.jpg
* fix mmdet torchscript
* fix test.py
* fix sdk forward
* fix yapf
2022-09-15 21:54:57 +08:00
966d737a1b
Fix: skip tests for uninstalled codebases ( #987 )
...
* skip tests if codebase not installed
* skip ort run test
* fix mmseg
2022-09-03 15:07:08 +08:00
0111e1f10f
[Feature] add swin for cls ( #911 )
...
* add swin for cls
* add ut and doc
* reduce trt batch size
* add regression test
* resolve comments
* remove useless rewriting logic
2022-08-23 17:14:24 +08:00
27a856637c
fix mmcls ut
2022-07-11 19:10:04 +08:00
17a7d60c7d
pass img_metas while exporting to onnx ( #681 )
...
* pass img_metas while exporting to onnx
* remove try-catch in tools for beter debugging
* use get
* fix typo
2022-06-30 17:33:24 +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
q.yao
DDGRCF
RunningLeon
hanrui1sensetime
AllentDan
grimoire
tpoisonooo
RunningLeon
Yifan Zhou