4dd4d4851b
Add rv1126 yolov3 support to sdk ( #1280 )
...
* add yolov3 head to SDK
* add yolov5 head to SDK
* fix export-info and lint, add reverse check
* fix lint
* fix export info for yolo heads
* add output_names to partition_config
* fix typo
* config
* normalize config
* fix
* refactor config
* fix lint and doc
* c++ form
* resolve comments
* fix CI
* fix CI
* fix CI
* float strides anchors
* refine pipeline of rknn-int8
* config
* rename func
* refactor
* rknn wrapper dict and fix typo
* rknn wrapper output update, mmcls use end2end type
* fix typo
2022-11-22 20:16:22 +08:00
5923054bb4
Add Core ML common configuration ( #1308 )
...
* Add CoreML instance segmentation configs
* Add common configuration for Core ML backend
* Fix pre-commit hook failures
2022-11-08 17:25:56 +08:00
50bd6b1703
improvement(visualize.py): headless enable inference ( #1041 )
...
* improvement(visualize.py): split render and display
* fix(visualize.py): skip render and save for headless device
* docs(deploy.py): update desc
2022-09-19 11:30:47 +08:00
124635ec5f
Add RKNN support. ( #865 )
...
* save codes
* support resnet and yolov3
* support yolox
* fix lint
* add mmseg support and a doc
* add UT
* update supported model list
* fix ci
* refine docstring
* resolve comments
* remote output_tensor_type
* resolve comments
* update readme
2022-09-06 11:48:39 +08:00
a0fb3be0df
support coreml ( #760 )
...
* sdk inference
* fix typo
* fix typo
* add convert things
* fix missling name
* add cls support
* add more pytorch rewriter
* add det support
* support det wip
* make Model export model_path
* fix nms
* add output back
* add docstring
* fix lint
* add coreml build action
* add zh docs
* add coreml backend check
* update ci
* update
* update
* update
* update
* update
* fix lint
* update configs
* add return value when error occured
* update docs
* update docs
* update docs
* fix lint
* udpate docs
* udpate docs
* update
Co-authored-by: grimoire <streetyao@live.com>
2022-09-05 19:55:47 +08:00
792c27b054
[Feature] Ascend backend ( #747 )
...
* add acl backend
* support dynamic batch size and dynamic image size
* add preliminary ascend backend
* support dtypes other than float
* support dynamic_dims in SDK
* fix dynamic batch size
* better error handling
* remove debug info
* [WIP] dynamic shape support
* fix static shape
* fix dynamic batch size
* add retinanet support
* fix dynamic image size
* fix dynamic image size
* fix dynamic dims
* fix dynamic dims
* simplify config files
* fix yolox support
* fix negative index
* support faster rcnn
* add seg config
* update benchmark
* fix onnx2ascend dynamic shape
* update docstring and benchmark
* add unit test, update documents
* fix wrapper
* fix ut
* fix for vit
* error handling
* context handling & multi-device support
* build with stub libraries
* add ci
* fix lint
* fix lint
* update doc ref
* fix typo
* down with `target_link_directories`
* setup python
* makedir
* fix ci
* fix ci
* remove verbose logs
* fix UBs
* export Error
* fix lint
* update checkenv
Co-authored-by: grimoire <yaoqian@sensetime.com>
2022-09-05 12:08:36 +08:00
cad0092801
[Fix] fix device error in dump-info ( #912 )
...
* fix device error in dump-info
* fix UT
2022-08-17 18:01:26 +08:00
80d24fc4da
fix error when set device cpu && fix docs error ( #866 )
2022-08-05 11:30:30 +08:00
2968a18d80
feat(tools/deploy.py): support snpe ( #789 )
...
* fix(tools/deploy.py): support snpe
* improvement(backend/snpe): review advices
* docs(backend/snpe): update build
* docs(backend/snpe): server support specify port
* docs(backend/snpe): update path
* fix(backend/snpe): time counter missing argument
* docs(backend/snpe): add missing argument
* docs(backend/snpe): update download and using
* improvement(snpe_net.cpp): load model with modeldata
2022-08-01 11:08:55 +08:00
41ddcafbb4
Rename onnx2ncnn to mmdeploy_onnx2ncnn ( #694 )
...
* improvement(tools/onnx2ncnn.py): rename to mmdeploy_onnx2ncnn
* format(tools/deploy.py): clean code
* fix(init_plugins.py): improve if condition
* fix(CI): update target
* fix(test_onnx2ncnn.py): update desc
* Update init_plugins.py
2022-07-08 09:21:32 +08:00
a228f95335
Improve ncnn patch embed ( #592 )
...
* improvement(PatchEmbed): use reshape instead of flatten
* improvement(init_plugins.py): get onnx2ncnn path
* fix(CI): build error
* fix(CI): revert get onnx2ncnn path
2022-06-15 19:16:51 +08:00
594c835b6b
merge master and fix the conflicts
2022-06-07 19:04:29 +08:00
b32fc41bed
[Refactor][API2.0] Api refactor2.0 ( #529 )
...
* [refactor][API2.0] Add onnx export and jit trace (#419 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add kwargs
* remove comment
* better pipeline manager
* remove print
* [Refactor][API2.0] Api partition calibration (#433 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add partition
* move calibration
* Better create_calib_table
* better deploy
* add kwargs
* remove comment
* better pipeline manager
* rename api, remove reduant variable, and misc
* [Refactor][API2.0] Api ncnn openvino (#435 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add ncnn api
* finish ncnn api
* add openvino support
* add kwargs
* remove comment
* better pipeline manager
* merge fix
* merge util and onnx2ncnn
* fix docstring
* [Refactor][API2.0] API for TensorRT (#519 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add partition
* move calibration
* Better create_calib_table
* better deploy
* add kwargs
* remove comment
* Add tensorrt API
* better pipeline manager
* add tensorrt new api
* remove print
* rename api, remove reduant variable, and misc
* add docstring
* [Refactor][API2.0] Api ppl other (#528 )
* first commit
* add async call
* add new api onnx export and jit trace
* add decorator
* fix ci
* fix torchscript ci
* fix loader
* better pipemanager
* remove comment, better import
* add kwargs
* Add new APIS for pplnn sdk and misc
* remove comment
* better pipeline manager
* merge fix
* update tools/onnx2pplnn.py
* rename function
2022-05-31 09:18:18 +08:00
ea0a9e5d43
feat(tools/deploy.py): support ncnn quantization ( #476 )
...
* feat(tools): add onnx2ncnn_quant_table
* feat(tools): add quantization image dataset
* feat(tools): add image dataset
* feat(tools/deploy.py): support quant
* fix(CI): lint
* fix(CI): format
* docs(zh_cn): add quantization usage
* docs(zh_cn): add benchmark
* feat(tools): add onnx2ncnn_quant_table
* docs(zh_cn): add more test result
* CI(github): add quant script
* CI(.github/scripts): add test quant
* fix(CI): remove pushd and popd
* feat(CI): debug
* fix(CI): path error
* fix(CI): fix path
* fix(CI): install wget
* fix(CI): review advices
* improvement(mmdeploy): review advice
* fix(tools): rename to onnx2ncnn_quant_table.py
* improvement(tools): rename file
* improvement(test): remove useless
* fix(tools/quant_image_dataset): remove loadFile in test.pipeline
* docs(quantization.md): update description
* fix(CI): protobuf version
* fix(CI): pip install
* docs(quantization): review advice
* fix(CI): revert mmcv version
* fix(CI): udpate pb version
* fix(CI): update
2022-05-26 19:53:56 +08:00
ba641c3b23
[Fix] set optional arg a default value ( #483 )
...
* optional default value
* resolve comments
Co-authored-by: dongchunyu.vendor <dongchunyu@pjlab.org.cn>
2022-05-17 19:57:12 +08:00
0cd44a6799
[Fix] Fix display bugs for windows ( #451 )
...
* fix issue 330 for windows
* fix code
* fix lint
* fix all platform
2022-05-12 12:00:57 +08:00
9306bcec80
Dev v0.4.0 ( #301 )
...
* bump version to v0.4.0
* [Enhancement] Make rewriter more powerful (#150 )
* Finish function tests
* lint
* resolve comments
* Fix tests
* docstring & fix
* Complement informations
* lint
* Add example
* Fix version
* Remove todo
Co-authored-by: RunningLeon <mnsheng@yeah.net>
* Torchscript support (#159 )
* support torchscript
* add nms
* add torchscript configs and update deploy process and dump-info
* typescript -> torchscript
* add torchscript custom extension support
* add ts custom ops again
* support mmseg unet
* [WIP] add optimizer for torchscript (#119 )
* add passes
* add python api
* Torchscript optimizer python api (#121 )
* add passes
* add python api
* use python api instead of executable
* Merge Master, update optimizer (#151 )
* [Feature] add yolox ncnn (#29 )
* add yolox ncnn
* add ncnn android performance of yolox
* add ut
* fix lint
* fix None bugs for ncnn
* test codecov
* test codecov
* add device
* fix yapf
* remove if-else for img shape
* use channelshuffle optimize
* change benchmark after channelshuffle
* fix yapf
* fix yapf
* fuse continuous reshape
* fix static shape deploy
* fix code
* drop pad
* only static shape
* fix static
* fix docstring
* Added mask overlay to output image, changed fprintf info messages to … (#55 )
* Added mask overlay to output image, changed fprintf info messages to stdout
* Improved box filtering (filter area/score), make sure roi coordinates stay within bounds
* clang-format
* Support UNet in mmseg (#77 )
* Repeatdataset in train has no CLASSES & PALETTE
* update result for unet
* update docstring for mmdet
* remove ppl for unet in docs
* fix ort wrap about input type (#81 )
* Fix memleak (#86 )
* delete []
* fix build error when enble MMDEPLOY_ACTIVE_LEVEL
* fix lint
* [Doc] Nano benchmark and tutorial (#71 )
* add cls benchmark
* add nano zh-cn benchmark and en tutorial
* add device row
* add doc path to index.rst
* fix typo
* [Fix] fix missing deploy_core (#80 )
* fix missing deploy_core
* mv flag to demo
* target link
* [Docs] Fix links in Chinese doc (#84 )
* Fix docs in Chinese link
* Fix links
* Delete symbolic link and add links to html
* delete files
* Fix link
* [Feature] Add docker files (#67 )
* add gpu and cpu dockerfile
* fix lint
* fix cpu docker and remove redundant
* use pip instead
* add build arg and readme
* fix grammar
* update readme
* add chinese doc for dockerfile and add docker build to build.md
* grammar
* refine dockerfiles
* add FAQs
* update Dpplcv_DIR for SDK building
* remove mmcls
* add sdk demos
* fix typo and lint
* update FAQs
* [Fix]fix check_env (#101 )
* fix check_env
* update
* Replace convert_syncbatchnorm in mmseg (#93 )
* replace convert_syncbatchnorm with revert_sync_batchnorm from mmcv
* change logger
* [Doc] Update FAQ for TensorRT (#96 )
* update FAQ
* comment
* [Docs]: Update doc for openvino installation (#102 )
* fix docs
* fix docs
* fix docs
* fix mmcv version
* fix docs
* rm blank line
* simplify non batch nms (#99 )
* [Enhacement] Allow test.py to save evaluation results (#108 )
* Add log file
* Delete debug code
* Rename logger
* resolve comments
* [Enhancement] Support mmocr v0.4+ (#115 )
* support mmocr v0.4+
* 0.4.0 -> 0.4.1
* fix onnxruntime wrapper for gpu inference (#123 )
* fix ncnn wrapper for ort-gpu
* resolve comment
* fix lint
* Fix typo (#132 )
* lock mmcls version (#131 )
* [Enhancement] upgrade isort in pre-commit config (#141 )
* [Enhancement] upgrade isort in pre-commit config by refering to mmflow pr #87
* fix lint
* remove .isort.cfg and put its known_third_party to setup.cfg
* Fix ci for mmocr (#144 )
* fix mmocr unittests
* remove useless
* lock mmdet maximum version to 2.20
* pip install -U numpy
* Fix capture_output (#125 )
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: Johannes L <tehkillerbee@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: 杨培文 (Yang Peiwen) <915505626@qq.com>
Co-authored-by: Semyon Bevzyuk <semen.bevzuk@gmail.com>
* configs for all tasks
* use torchvision roi align
* remote unnecessary code
* fix ut
* fix ut
* export
* det dynamic
* det dynamic
* add ut
* fix ut
* add ut and docs
* fix ut
* skip torchscript ut if no ops available
* add torchscript option to build.md
* update benchmark and resolve comments
* resolve conflicts
* rename configs
* fix mrcnn cuda test
* remove useless
* add version requirements to docs and comments to codes
* enable empty image exporting for torchscript and accelerate ORT inference for MRCNN
* rebase
* update example for torchscript.md
* update FAQs for torchscript.md
* resolve comments
* only use torchvision roi_align for torchscript
* fix ut
* use torchvision roi align when pool model is avg
* resolve comments
Co-authored-by: grimoire <streetyao@live.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: Johannes L <tehkillerbee@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: 杨培文 (Yang Peiwen) <915505626@qq.com>
Co-authored-by: Semyon Bevzyuk <semen.bevzuk@gmail.com>
* Update supported mmseg models (#181 )
* fix ocrnet cascade decoder
* update mmseg support models
* update mmseg configs
* support emanet and icnet
* set max K of TopK for tensorrt
* update supported models for mmseg in docs
* add test for emamodule
* add configs and update docs
* Update docs
* update benchmark
* [Features]Support mmdet3d (#103 )
* add mmdet3d code
* add code
* update code
* [log]This commit finish pointpillar export and evaluate on onnxruntime.The model is sample with nvidia repo model
* add tensorrt config
* fix config
* update
* support for tensorrt
* add config
* fix config`
* fix apis about torch2onnx
* update
* mmdet3d deploy version1.0
* map is ok
* fix code
* version1.0
* fix code
* fix visual
* fix bug
* tensorrt support success
* add docstring
* add docs
* fix docs
* fix comments
* fix comment
* fix comment
* fix openvino wrapper
* add unit test
* fix device about cpu
* fix comment
* fix show_result
* fix lint
* fix requirments
* remove ci about det3d
* fix ut
* add ut data
* support for new version pointpillars
* fix comment
* fix support_list
* fix comments
* fix config name
* [Enhancement] Update pad logic in detection heads (#168 )
* pad with register
* fix lint
Co-authored-by: AllentDan <dongchunyu@sensetime.com>
* [Enhancement] Additional arguments support for OpenVINO Model Optimizer (#178 )
* Add mo args.
* [Docs]: update docs and argument descriptions (#196 )
* bump version to v0.4.0
* update docs and argument descriptions
* revert version change
* fix unnecessary change of config for dynamic exportation (#199 )
* fix mmcls get classes (#215 )
* fix mmcls get classes
* resolve comment
* resolve comment
* Add ModelOptimizerOptions.
* Fix merge bugs.
* Update mmpose.md (#224 )
* [Dostring]add example in apis docstring (#214 )
* add example in apis docstring
* add backend example in docstring
* rm blank line
* Fixed get_mo_options_from_cfg args
* fix l2norm test
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: Haofan Wang <frankmiracle@outlook.com>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
* [Enhancement] Switch to statically typed Value::Any (#209 )
* replace std::any with StaticAny
* fix __compare_typeid
* remove fallback id support
* constraint on traits::TypeId<T>::value
* fix includes
* [Enhancement] TensorRT DCN support (#205 )
* add tensorrt dcn support
* fix lint
* remove roi_align plugin for ORT (#258 )
* remove roi_align plugin
* remove ut
* skip single_roi_extractor UT for ORT in CI
* move align to symbolic and update docs
* recover UT
* resolve comments
* [Enhancement]: Support fcn_unet deployment with dynamic shape (#251 )
* support mmseg fcn+unet dynamic shape
* add test
* fix ci
* fix units
* resolve comments
* [Enhancement] fix-cmake-relocatable (#223 )
* require user to specify xxx_dir
* fix line ending
* fix end-of-file-fixer
* try to fix ld cudart cublas
* add ENV var search
* fix CMAKE_CUDA_COMPILER
* cpu, cuda should all work well
* remove commented code
* fix ncnn example find ncnn package (#282 )
* table format is wrong (#283 )
* update pre-commit (#284 )
* update pre-commit
* fix clang-format
* fix mmseg config (#281 )
* fix mmseg config
* fix mmpose evaluate outputs
* fix lint
* update pre-commit config
* fix lint
* Revert "update pre-commit config"
This reverts commit c3fd71611f#296 )
* refactor cmake build (#295 )
* add-mmpose-sdk (#259 )
* Torchscript support (#159 )
* support torchscript
* add nms
* add torchscript configs and update deploy process and dump-info
* typescript -> torchscript
* add torchscript custom extension support
* add ts custom ops again
* support mmseg unet
* [WIP] add optimizer for torchscript (#119 )
* add passes
* add python api
* Torchscript optimizer python api (#121 )
* add passes
* add python api
* use python api instead of executable
* Merge Master, update optimizer (#151 )
* [Feature] add yolox ncnn (#29 )
* add yolox ncnn
* add ncnn android performance of yolox
* add ut
* fix lint
* fix None bugs for ncnn
* test codecov
* test codecov
* add device
* fix yapf
* remove if-else for img shape
* use channelshuffle optimize
* change benchmark after channelshuffle
* fix yapf
* fix yapf
* fuse continuous reshape
* fix static shape deploy
* fix code
* drop pad
* only static shape
* fix static
* fix docstring
* Added mask overlay to output image, changed fprintf info messages to … (#55 )
* Added mask overlay to output image, changed fprintf info messages to stdout
* Improved box filtering (filter area/score), make sure roi coordinates stay within bounds
* clang-format
* Support UNet in mmseg (#77 )
* Repeatdataset in train has no CLASSES & PALETTE
* update result for unet
* update docstring for mmdet
* remove ppl for unet in docs
* fix ort wrap about input type (#81 )
* Fix memleak (#86 )
* delete []
* fix build error when enble MMDEPLOY_ACTIVE_LEVEL
* fix lint
* [Doc] Nano benchmark and tutorial (#71 )
* add cls benchmark
* add nano zh-cn benchmark and en tutorial
* add device row
* add doc path to index.rst
* fix typo
* [Fix] fix missing deploy_core (#80 )
* fix missing deploy_core
* mv flag to demo
* target link
* [Docs] Fix links in Chinese doc (#84 )
* Fix docs in Chinese link
* Fix links
* Delete symbolic link and add links to html
* delete files
* Fix link
* [Feature] Add docker files (#67 )
* add gpu and cpu dockerfile
* fix lint
* fix cpu docker and remove redundant
* use pip instead
* add build arg and readme
* fix grammar
* update readme
* add chinese doc for dockerfile and add docker build to build.md
* grammar
* refine dockerfiles
* add FAQs
* update Dpplcv_DIR for SDK building
* remove mmcls
* add sdk demos
* fix typo and lint
* update FAQs
* [Fix]fix check_env (#101 )
* fix check_env
* update
* Replace convert_syncbatchnorm in mmseg (#93 )
* replace convert_syncbatchnorm with revert_sync_batchnorm from mmcv
* change logger
* [Doc] Update FAQ for TensorRT (#96 )
* update FAQ
* comment
* [Docs]: Update doc for openvino installation (#102 )
* fix docs
* fix docs
* fix docs
* fix mmcv version
* fix docs
* rm blank line
* simplify non batch nms (#99 )
* [Enhacement] Allow test.py to save evaluation results (#108 )
* Add log file
* Delete debug code
* Rename logger
* resolve comments
* [Enhancement] Support mmocr v0.4+ (#115 )
* support mmocr v0.4+
* 0.4.0 -> 0.4.1
* fix onnxruntime wrapper for gpu inference (#123 )
* fix ncnn wrapper for ort-gpu
* resolve comment
* fix lint
* Fix typo (#132 )
* lock mmcls version (#131 )
* [Enhancement] upgrade isort in pre-commit config (#141 )
* [Enhancement] upgrade isort in pre-commit config by refering to mmflow pr #87
* fix lint
* remove .isort.cfg and put its known_third_party to setup.cfg
* Fix ci for mmocr (#144 )
* fix mmocr unittests
* remove useless
* lock mmdet maximum version to 2.20
* pip install -U numpy
* Fix capture_output (#125 )
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: Johannes L <tehkillerbee@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: 杨培文 (Yang Peiwen) <915505626@qq.com>
Co-authored-by: Semyon Bevzyuk <semen.bevzuk@gmail.com>
* configs for all tasks
* use torchvision roi align
* remote unnecessary code
* fix ut
* fix ut
* export
* det dynamic
* det dynamic
* add ut
* fix ut
* add ut and docs
* fix ut
* skip torchscript ut if no ops available
* add torchscript option to build.md
* update benchmark and resolve comments
* resolve conflicts
* rename configs
* fix mrcnn cuda test
* remove useless
* add version requirements to docs and comments to codes
* enable empty image exporting for torchscript and accelerate ORT inference for MRCNN
* rebase
* update example for torchscript.md
* update FAQs for torchscript.md
* resolve comments
* only use torchvision roi_align for torchscript
* fix ut
* use torchvision roi align when pool model is avg
* resolve comments
Co-authored-by: grimoire <streetyao@live.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: Johannes L <tehkillerbee@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: 杨培文 (Yang Peiwen) <915505626@qq.com>
Co-authored-by: Semyon Bevzyuk <semen.bevzuk@gmail.com>
* Update supported mmseg models (#181 )
* fix ocrnet cascade decoder
* update mmseg support models
* update mmseg configs
* support emanet and icnet
* set max K of TopK for tensorrt
* update supported models for mmseg in docs
* add test for emamodule
* add configs and update docs
* Update docs
* update benchmark
* [Features]Support mmdet3d (#103 )
* add mmdet3d code
* add code
* update code
* [log]This commit finish pointpillar export and evaluate on onnxruntime.The model is sample with nvidia repo model
* add tensorrt config
* fix config
* update
* support for tensorrt
* add config
* fix config`
* fix apis about torch2onnx
* update
* mmdet3d deploy version1.0
* map is ok
* fix code
* version1.0
* fix code
* fix visual
* fix bug
* tensorrt support success
* add docstring
* add docs
* fix docs
* fix comments
* fix comment
* fix comment
* fix openvino wrapper
* add unit test
* fix device about cpu
* fix comment
* fix show_result
* fix lint
* fix requirments
* remove ci about det3d
* fix ut
* add ut data
* support for new version pointpillars
* fix comment
* fix support_list
* fix comments
* fix config name
* [Enhancement] Additional arguments support for OpenVINO Model Optimizer (#178 )
* Add mo args.
* [Docs]: update docs and argument descriptions (#196 )
* bump version to v0.4.0
* update docs and argument descriptions
* revert version change
* fix unnecessary change of config for dynamic exportation (#199 )
* fix mmcls get classes (#215 )
* fix mmcls get classes
* resolve comment
* resolve comment
* Add ModelOptimizerOptions.
* Fix merge bugs.
* Update mmpose.md (#224 )
* [Dostring]add example in apis docstring (#214 )
* add example in apis docstring
* add backend example in docstring
* rm blank line
* Fixed get_mo_options_from_cfg args
* fix l2norm test
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: Haofan Wang <frankmiracle@outlook.com>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
* add-mmpose-codebase
* fix ci
* fix img_shape after TopDownAffine
* rename TopDown module -> XheadDecode & implement regression decode
* align keypoints_from_heatmap
* remove hardcode keypoint_head, need refactor, current only support topdown config
* add mmpose python api
* update mmpose-python code
* can't clip fake box
* fix rebase error
* fix rebase error
* link mspn decoder to base decoder
* fix ci
* compile with gcc7.5
* remove no use code
* fix
* fix prompt
* remove unnecessary cv::parallel_for_
* rewrite TopdownHeatmapMultiStageHead.inference_model
* add comment
* add more detail docstring why use _cs2xyxy in sdk backend
* fix Registry name
* remove no use param & add comment of output result
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: grimoire <streetyao@live.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: Johannes L <tehkillerbee@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: 杨培文 (Yang Peiwen) <915505626@qq.com>
Co-authored-by: Semyon Bevzyuk <semen.bevzuk@gmail.com>
Co-authored-by: Haofan Wang <frankmiracle@outlook.com>
* update faq about WinError 1455 (#297 )
* update faq about WinError 1455
* Update faq.md
* Update faq.md
* fix ci
Co-authored-by: chenxin2 <chenxin2@sensetime.com>
* [Feature]Support centerpoint (#252 )
* bump version to v0.4.0
* [Enhancement] Make rewriter more powerful (#150 )
* Finish function tests
* lint
* resolve comments
* Fix tests
* docstring & fix
* Complement informations
* lint
* Add example
* Fix version
* Remove todo
Co-authored-by: RunningLeon <mnsheng@yeah.net>
* Torchscript support (#159 )
* support torchscript
* add nms
* add torchscript configs and update deploy process and dump-info
* typescript -> torchscript
* add torchscript custom extension support
* add ts custom ops again
* support mmseg unet
* [WIP] add optimizer for torchscript (#119 )
* add passes
* add python api
* Torchscript optimizer python api (#121 )
* add passes
* add python api
* use python api instead of executable
* Merge Master, update optimizer (#151 )
* [Feature] add yolox ncnn (#29 )
* add yolox ncnn
* add ncnn android performance of yolox
* add ut
* fix lint
* fix None bugs for ncnn
* test codecov
* test codecov
* add device
* fix yapf
* remove if-else for img shape
* use channelshuffle optimize
* change benchmark after channelshuffle
* fix yapf
* fix yapf
* fuse continuous reshape
* fix static shape deploy
* fix code
* drop pad
* only static shape
* fix static
* fix docstring
* Added mask overlay to output image, changed fprintf info messages to … (#55 )
* Added mask overlay to output image, changed fprintf info messages to stdout
* Improved box filtering (filter area/score), make sure roi coordinates stay within bounds
* clang-format
* Support UNet in mmseg (#77 )
* Repeatdataset in train has no CLASSES & PALETTE
* update result for unet
* update docstring for mmdet
* remove ppl for unet in docs
* fix ort wrap about input type (#81 )
* Fix memleak (#86 )
* delete []
* fix build error when enble MMDEPLOY_ACTIVE_LEVEL
* fix lint
* [Doc] Nano benchmark and tutorial (#71 )
* add cls benchmark
* add nano zh-cn benchmark and en tutorial
* add device row
* add doc path to index.rst
* fix typo
* [Fix] fix missing deploy_core (#80 )
* fix missing deploy_core
* mv flag to demo
* target link
* [Docs] Fix links in Chinese doc (#84 )
* Fix docs in Chinese link
* Fix links
* Delete symbolic link and add links to html
* delete files
* Fix link
* [Feature] Add docker files (#67 )
* add gpu and cpu dockerfile
* fix lint
* fix cpu docker and remove redundant
* use pip instead
* add build arg and readme
* fix grammar
* update readme
* add chinese doc for dockerfile and add docker build to build.md
* grammar
* refine dockerfiles
* add FAQs
* update Dpplcv_DIR for SDK building
* remove mmcls
* add sdk demos
* fix typo and lint
* update FAQs
* [Fix]fix check_env (#101 )
* fix check_env
* update
* Replace convert_syncbatchnorm in mmseg (#93 )
* replace convert_syncbatchnorm with revert_sync_batchnorm from mmcv
* change logger
* [Doc] Update FAQ for TensorRT (#96 )
* update FAQ
* comment
* [Docs]: Update doc for openvino installation (#102 )
* fix docs
* fix docs
* fix docs
* fix mmcv version
* fix docs
* rm blank line
* simplify non batch nms (#99 )
* [Enhacement] Allow test.py to save evaluation results (#108 )
* Add log file
* Delete debug code
* Rename logger
* resolve comments
* [Enhancement] Support mmocr v0.4+ (#115 )
* support mmocr v0.4+
* 0.4.0 -> 0.4.1
* fix onnxruntime wrapper for gpu inference (#123 )
* fix ncnn wrapper for ort-gpu
* resolve comment
* fix lint
* Fix typo (#132 )
* lock mmcls version (#131 )
* [Enhancement] upgrade isort in pre-commit config (#141 )
* [Enhancement] upgrade isort in pre-commit config by refering to mmflow pr #87
* fix lint
* remove .isort.cfg and put its known_third_party to setup.cfg
* Fix ci for mmocr (#144 )
* fix mmocr unittests
* remove useless
* lock mmdet maximum version to 2.20
* pip install -U numpy
* Fix capture_output (#125 )
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: Johannes L <tehkillerbee@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: 杨培文 (Yang Peiwen) <915505626@qq.com>
Co-authored-by: Semyon Bevzyuk <semen.bevzuk@gmail.com>
* configs for all tasks
* use torchvision roi align
* remote unnecessary code
* fix ut
* fix ut
* export
* det dynamic
* det dynamic
* add ut
* fix ut
* add ut and docs
* fix ut
* skip torchscript ut if no ops available
* add torchscript option to build.md
* update benchmark and resolve comments
* resolve conflicts
* rename configs
* fix mrcnn cuda test
* remove useless
* add version requirements to docs and comments to codes
* enable empty image exporting for torchscript and accelerate ORT inference for MRCNN
* rebase
* update example for torchscript.md
* update FAQs for torchscript.md
* resolve comments
* only use torchvision roi_align for torchscript
* fix ut
* use torchvision roi align when pool model is avg
* resolve comments
Co-authored-by: grimoire <streetyao@live.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: Johannes L <tehkillerbee@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: 杨培文 (Yang Peiwen) <915505626@qq.com>
Co-authored-by: Semyon Bevzyuk <semen.bevzuk@gmail.com>
* Update supported mmseg models (#181 )
* fix ocrnet cascade decoder
* update mmseg support models
* update mmseg configs
* support emanet and icnet
* set max K of TopK for tensorrt
* update supported models for mmseg in docs
* add test for emamodule
* add configs and update docs
* Update docs
* update benchmark
* [Features]Support mmdet3d (#103 )
* add mmdet3d code
* add code
* update code
* [log]This commit finish pointpillar export and evaluate on onnxruntime.The model is sample with nvidia repo model
* add tensorrt config
* fix config
* update
* support for tensorrt
* add config
* fix config`
* fix apis about torch2onnx
* update
* mmdet3d deploy version1.0
* map is ok
* fix code
* version1.0
* fix code
* fix visual
* fix bug
* tensorrt support success
* add docstring
* add docs
* fix docs
* fix comments
* fix comment
* fix comment
* fix openvino wrapper
* add unit test
* fix device about cpu
* fix comment
* fix show_result
* fix lint
* fix requirments
* remove ci about det3d
* fix ut
* add ut data
* support for new version pointpillars
* fix comment
* fix support_list
* fix comments
* fix config name
* [Enhancement] Update pad logic in detection heads (#168 )
* pad with register
* fix lint
Co-authored-by: AllentDan <dongchunyu@sensetime.com>
* [Enhancement] Additional arguments support for OpenVINO Model Optimizer (#178 )
* Add mo args.
* [Docs]: update docs and argument descriptions (#196 )
* bump version to v0.4.0
* update docs and argument descriptions
* revert version change
* fix unnecessary change of config for dynamic exportation (#199 )
* fix mmcls get classes (#215 )
* fix mmcls get classes
* resolve comment
* resolve comment
* Add ModelOptimizerOptions.
* Fix merge bugs.
* Update mmpose.md (#224 )
* [Dostring]add example in apis docstring (#214 )
* add example in apis docstring
* add backend example in docstring
* rm blank line
* Fixed get_mo_options_from_cfg args
* fix l2norm test
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: Haofan Wang <frankmiracle@outlook.com>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
* [Enhancement] Switch to statically typed Value::Any (#209 )
* replace std::any with StaticAny
* fix __compare_typeid
* remove fallback id support
* constraint on traits::TypeId<T>::value
* fix includes
* support for centerpoint
* [Enhancement] TensorRT DCN support (#205 )
* add tensorrt dcn support
* fix lint
* add docstring and dcn model support
* add centerpoint ut and docs
* add config and fix input rank
* fix merge error
* fix a bug
* fix comment
* [Doc] update benchmark add supported-model-list (#286 )
* update benchmark add supported-model-list
* fix lint
* fix lint
* loc mmocr maximum version
* fix ut
Co-authored-by: maningsheng <mnsheng@yeah.net>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: grimoire <streetyao@live.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: Johannes L <tehkillerbee@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: 杨培文 (Yang Peiwen) <915505626@qq.com>
Co-authored-by: Semyon Bevzyuk <semen.bevzuk@gmail.com>
Co-authored-by: AllentDan <dongchunyu@sensetime.com>
Co-authored-by: Haofan Wang <frankmiracle@outlook.com>
Co-authored-by: lzhangzz <lzhang329@gmail.com>
Co-authored-by: maningsheng <mnsheng@yeah.net>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: grimoire <streetyao@live.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: Johannes L <tehkillerbee@users.noreply.github.com>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: 杨培文 (Yang Peiwen) <915505626@qq.com>
Co-authored-by: Semyon Bevzyuk <semen.bevzuk@gmail.com>
Co-authored-by: AllentDan <dongchunyu@sensetime.com>
Co-authored-by: Haofan Wang <frankmiracle@outlook.com>
Co-authored-by: lzhangzz <lzhang329@gmail.com>
Co-authored-by: Chen Xin <xinchen.tju@gmail.com>
Co-authored-by: chenxin2 <chenxin2@sensetime.com>
2022-04-01 18:14:23 +08:00
47b8e6e71c
[Doc] update benchmark add supported-model-list ( #286 )
...
* update benchmark add supported-model-list
* fix lint
* fix lint
* loc mmocr maximum version
2022-04-01 12:26:34 +08:00
3fa2cc8845
[Fix] add support for headless device ( #250 )
...
* add support for headless device
* move headless code
* fix lint
* add no $DISPLAY warning
* add some warnings
* fix info
2022-03-25 09:53:03 +08:00
8552d35ceb
[Docs]: update docs and argument descriptions ( #196 )
...
* bump version to v0.4.0
* update docs and argument descriptions
* revert version change
2022-03-08 16:53:11 +08:00
be93956015
fix tools ( #160 )
...
* add non-zero error code
* update mmdeploy version
2022-02-16 11:50:04 +08:00
d157243077
[Fix] move target_wrapper into utils ( #20 )
...
* move target_wrapper into utils
* fix for lint
* add typehint and docstring
* update unit test
* fix isort
* update import
2022-01-11 15:43:47 +08:00
dbc12686ea
[Enhancement] use mmcv.utils.get_logger ( #24 )
...
* use mmcv.utils.get_logger
* add a unit test
* fix ut
* refine docstring
* fix lint
2022-01-10 15:46:42 +08:00
8cdb4abcc9
[Fix] Fix input names ( #12 )
...
* [Fix] fix bugs for mmcls performance test (#269 )
* fix bugs for mmcls performance test
* fix yapf
* add comments of CLASSES attribute
* Fix input name
* support all codebases and update dump info
* fix docstring
* Added random names.
* onnx_config -> ir_config
* fix test
* fix test
* add a TODO
* use self.output_names
* Fixes
* Rewrite test_torch2onnx.
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: AllentDan <AllentDan@yeah.net>
Co-authored-by: grimoire <yaoqian@sensetime.com>
2021-12-30 19:56:03 +08:00
0f9f0a60e4
[Enhancement]: Update configs for OpenVINO ( #306 )
...
* Update config for OpenVINO
* Update test
* Update better test
* Update instance-seg config of OpenVINO
2021-12-20 17:56:25 +08:00
f94d7b5abe
[Feature] Dump jsons to SDK ( #279 )
...
* save cods
* save codes
* save codes
* add docstring and fix ut
* use Resize to replace img_scale
* use Dict instead of dict()
* net && weights
* resolve comments
* update pipeline
* update pipeline
* update pipeline
* refactor preprocess
* merge master
* use imagetotensor
* use imagetotensor
* fix typos and improve docstring
* resolve comments
* update lq to img in mmedit
* add mask_thr_binary
* refactor dump info
* resolve comments
* add instance seg to dump info temperarily
* update meta_keys
* update meta_keys
* add meta_keys condition
* update postprocess for mmseg
2021-12-17 11:39:16 +08:00
0dea300714
[Fix] Reanme ppl to pplnn for all cases ( #273 )
...
* add speed benchmark for pplnn and replace all ppl in docs with pplnn
* replace ppl with pplnn for all codes
* docs zh
* fix crnn config path
* update docstring
* rename
* update ppl to pplnn in readme_zh
Co-authored-by: maningsheng <mnsheng@yeah.net>
2021-12-14 15:04:39 +08:00
03c95a1149
[Enhancement]: Optimize config utils ( #263 )
...
* Optimize config utils
* Update `get_backend`
* Add assert
2021-12-08 14:55:59 +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
a8abc75065
Add licence head to code ( #236 )
2021-11-30 15:00:37 +08:00
d0bb497b0c
[Fix] fix ncnn apis, add onnx2ncnn unit tests, and rename test_mmseg/test_utils.py to avoid conflicts ( #233 )
...
* refactor ncnn convert
* refine docstring
* add onnx2ncnn test
* resolve comments and rename test_utils.py in test_mmseg
Co-authored-by: maningsheng <mnsheng@yeah.net>
2021-11-26 11:43:27 +08:00
3a785f1223
[Refactor] Refactor codebase ( #220 )
...
* [WIP] Refactor v2.0 (#163 )
* Refactor backend wrapper
* Refactor mmdet.inference
* Fix
* merge
* refactor utils
* Use deployer and deploy_model to manage pipeline
* Resolve comments
* Add a real inference api function
* rename wrappers
* Set execute to private method
* Rename deployer deploy_model
* Refactor task
* remove type hint
* lint
* Resolve comments
* resolve comments
* lint
* docstring
* [Fix]: Fix bugs in details in refactor branch (#192 )
* [WIP] Refactor v2.0 (#163 )
* Refactor backend wrapper
* Refactor mmdet.inference
* Fix
* merge
* refactor utils
* Use deployer and deploy_model to manage pipeline
* Resolve comments
* Add a real inference api function
* rename wrappers
* Set execute to private method
* Rename deployer deploy_model
* Refactor task
* remove type hint
* lint
* Resolve comments
* resolve comments
* lint
* docstring
* Fix errors
* lint
* resolve comments
* fix bugs
* conflict
* lint and typo
* Resolve comment
* refactor mmseg (#201 )
* support mmseg
* fix docstring
* fix docstring
* [Refactor]: Get the count of backend files (#202 )
* Fix backend files
* resolve comments
* lint
* Fix ncnn
* [Refactor]: Refactor folders of mmdet (#200 )
* Move folders
* lint
* test object detection model
* lint
* reset changes
* fix openvino
* resolve comments
* __init__.py
* Fix path
* [Refactor]: move mmseg (#206 )
* [Refactor]: Refactor mmedit (#205 )
* feature mmedit
* edit2.0
* edit
* refactor mmedit
* fix __init__.py
* fix __init__
* fix formai
* fix comment
* fix comment
* Fix wrong func_name of ConvFCBBoxHead (#209 )
* [Refactor]: Refactor mmdet unit test (#207 )
* Move folders
* lint
* test object detection model
* lint
* WIP
* remove print
* finish unit test
* Fix tests
* resolve comments
* Add mask test
* lint
* resolve comments
* Refine cfg file
* Move files
* add files
* Fix path
* [Unittest]: Refine the unit tests in mmdet #214
* [Refactor] refactor mmocr to mmdeploy/codebase (#213 )
* refactor mmocr to mmdeploy/codebase
* fix docstring of show_result
* fix docstring of visualize
* refine docstring
* replace print with logging
* refince codes
* resolve comments
* resolve comments
* [Refactor]: mmseg tests (#210 )
* refactor mmseg tests
* rename test_codebase
* update
* add model.py
* fix
* [Refactor] Refactor mmcls and the package (#217 )
* refactor mmcls
* fix yapf
* fix isort
* refactor-mmcls-package
* fix print to logging
* fix docstrings according to others comments
* fix comments
* fix comments
* fix allentdans comment in pr215
* remove mmocr init
* [Refactor] Refactor mmedit tests (#212 )
* feature mmedit
* edit2.0
* edit
* refactor mmedit
* fix __init__.py
* fix __init__
* fix formai
* fix comment
* fix comment
* buff
* edit test and code refactor
* refactor dir
* refactor tests/mmedit
* fix docstring
* add test coverage
* fix lint
* fix comment
* fix comment
* Update typehint (#216 )
* update type hint
* update docstring
* update
* remove file
* fix ppl
* Refine get_predefined_partition_cfg
* fix tensorrt version > 8
* move parse_cuda_device_id to device.py
* Fix cascade
* onnx2ncnn docstring
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: RunningLeon <maningsheng@sensetime.com>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
2021-11-25 09:57:05 +08:00
7b369e0628
Update log format ( #203 )
2021-11-22 20:41:17 +08:00
a4dceb4bb4
[Enhancement] Enhance ppl for all codebases ( #177 )
...
* enhance ppl for all codebases
* fix dump info
* fix md and use not None
* remove redundant codes
* safe convert empty ppl tensor
* add examples and remove useless lines
2021-11-16 19:16:46 +08:00
c5a87fb1bc
[Enhancement]: OpenVINO deployment support for some models ( #117 )
...
* Fix include and lib paths for onnxruntime.
* Fixes for SSD export test
* Add onnx2openvino and OpenVINODetector. Test models: ssd, retinanet, fcos, fsaf.
* Add support for two-stage models: faster_rcnn, cascade_rcnn
* Add doc
* Add strip_doc_string for openvino.
* Fix openvino preprocess.
* Add OpenVINO to test_wrapper.py.
* Fix
* Add openvino_execute.
* Removed preprocessing.
* Fix onnxruntime cmake.
* Rewrote postprocessing and forward, added docstrings and fixes.
* Added device type change to OpenVINOWrapper.
* Update forward_of_single_roi_extractor_dynamic_openvino and fix doc.
* Update docs.
* Add OpenVINODetector and onn2openvino tests.
* Add input_info to onnx2openvino.
* Add TestOpenVINOExporter and test_single_roi_extractor.
* Moved get_input_shape_from_cfg to openvino_utils.py and added test.
* Added test_cascade_roi_head.
* Add backend.check_env() to tests.
* Add OpenVINO to get_rewrite_outputs and to some tests in test_mmdet_models.
* Moved test_single_roi_extractor to test_mmdet_models.
* Removed TestOpenVINOExporter.
2021-10-21 16:07:35 +08:00
5b8750b83b
[Enhance]: Logging trace info for debugging in deploy tool ( #93 )
...
* print trace info for debugging
* resolve comment
2021-09-24 19:09:39 +08:00
4c0b36b7ff
[Refactor] Refactor config v1 ( #80 )
...
* [Refactor] Refactor configs according to new standard (#67 )
* modify cfg and cfg_util
* modify tensorrt config
* fix bug
* lint
* Fix
1. Delete print
2. Modify the return value from "False, None" to "None" and related code
3. Rename 2 get functions
* modify apply_marks
* [Feature] Refactor ocr config (#71 )
* add text detection config refactor
* add text recognition refactor
* add static exporting for mmocr
* fix lint
* set max space in child config
* use Sequence[int] instead
* add assert input_shape
* fix static bug and add ppl ort and trt static (#77 )
* [Feature] Refine setup.py (#61 )
* add setup.py and related files
* lint
* Edit requirements
* modify onnx version
* modify according to comments
* [Refactor] Refactor mmseg configs (#73 )
* refactor mmseg config
* change create_input
* fix lint
* fix lint
* fix lint
* fix yapf
* fix yapf
* update export
* remove Segmentation
* remove tast assert
* add onnx_config
* remove hardcode
* Inherit with static
* Remove blank line
* Add segmentation task enum
* add assert task
* mmocr version 0.3.0 (#79 )
* add dump_info
* [Feature]: Refactor config in mmdet (#75 )
* support onnxruntime
* add two stage
* test two-stage ort and ppl
* update fcos post_params
* fix calib
* test ok with maskrcnn dynamic
* add empty line
* add static into config filename
* add input_shape to create_input in mmdet
* add static to some configs
* remove todo codes
* remove partition config in base
* refactor create_input
* rename task name in mmdet
* return None if input_shape is None
* add size info into mmdet configs filenames
* reorganize mmdet configs
* add object detection task for mmdet
* rename get_mmdet_params
* keep naming style consistent
* update post_params for fcos
* fix typo in ncnn config
* [Refactor] Refactor mmedit static config (#78 )
* add static cfg
* update create_input
* [Refactor]: Refactor mmcls configs (#74 )
* refactor mmcls2.0
* fix classify_tensorrt_dynamic.py
* fix classify_tensorrt_dynmic.py
* classify_tensorrt_dynamic_int8.py
* fix file name
* fix ncnn ppl
* updata prepare_input.py
* update utils.py
* updata constant.py
* add
* fix prepare_input.py
* fix prepare_input.py
* add static config file
* add blank lines
* fix prepare_input.py(wait test)
* fix input_shape(wait test)
* Update prepare_input.py
* fix classification_tensorrt_dynamic(wait test)
* fix classification_tensorrt_dynamic_int8(wait test)
* fix classification_tensorrt_static_int8(wait test)
* Rename classification_tensorrt_dynamic.py to classification_tensorrt_dynamic-224x224-224x224.py
* Rename classification_tensorrt_dynamic_int8.py to classification_tensorrt_dynamic_int8-224x224-224x224.py
* Rename classification_tensorrt_dynamic_int8-224x224-224x224.py to classification_tensorrt_int8_dynamic_224x224-224x224.py
* Rename classification_tensorrt_dynamic-224x224-224x224.py to classification_tensorrt_dynamic_224x224-224x224.py
* Rename classification_tensorrt_static.py to classification_tensorrt_static_224x224.py
* Rename classification_tensorrt_static_int8.py to classification_tensorrt_int8_static_224x224.py
* Update prepare_input.py
* Rename classification_tensorrt_dynamic_224x224-224x224.py to classification_tensorrt_dynamic-224x224-224x224.py
* Rename classification_tensorrt_int8_dynamic_224x224-224x224.py to classification_tensorrt_int8-dynamic_224x224-224x224.py
* Rename classification_tensorrt_int8-dynamic_224x224-224x224.py to classification_tensorrt_int8_dynamic-224x224-224x224.py
* Rename classification_tensorrt_int8_static_224x224.py to classification_tensorrt_int8_static-224x224.py
* Rename classification_tensorrt_static_224x224.py to classification_tensorrt_static-224x224.py
* Update prepare_input.py
* Update prepare_input.py
* Update prepare_input.py
* Update prepare_input.py
* Update prepare_input.py
* Update prepare_input.py
* Update prepare_input.py
* change logging msg
Co-authored-by: maningsheng <mnsheng@yeah.net>
* fix
* fix else branch
* fix bug for trt in mmseg
* enable dump trt info
* fix trt static for mmdet
* remove two-stage_partition_tensorrt_static-800x1344 config
* fix wrong backend in ppl config
* fix partition calibration
Co-authored-by: Yifan Zhou <singlezombie@163.com>
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: RunningLeon <maningsheng@sensetime.com>
Co-authored-by: VVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: maningsheng <mnsheng@yeah.net>
Co-authored-by: AllentDan <AllentDan@yeah.net>
2021-09-16 10:26:09 +08:00
10793f488e
[Feture] Export preprocess and deploy information to SDK ( #65 )
...
* add export info
* add dump-info funciton
* add collect info
* fix lint
* add docstring
* docstring
* docstring
2021-09-13 18:50:29 +08:00
15b25e2cd8
value=-1 as default ( #59 )
2021-09-06 10:27:53 +08:00
2b98040b26
[Feature] Refactor v1 ( #56 )
...
* [Refactor] add enum class and use functions to get configuration (#40 )
* add task and codebase enum class
* use funcitons to get config
* Refactor wrappers of mmcls and mmseg (#41 )
* move wrappers of cls & det to apis
* remove get_classes_from_config
* rename onnx_helper to onnx_utils
* move import to outside of class
* refactor ortwrappers
* Refactor build dataset and dataloader for mmseg (#44 )
* refactor build_dataset and build_dataloader for mmcls and mmseg
* remove repeated classes
* set build_dataloader with shuffle=False
* [Refactor] pplwrapper and mmocr refactor (#46 )
* add
* add pplwrapper and refactor mmocr
* fix lint
* remove unused arguments
* apply dict input for pplwrapper and ortwrapper
* add condition before import ppl and ort stuff
* update ppl (#51 )
* Refactor return value and extract_model (#54 )
* remove ret_value
* refactor extract_model
* fix typo
* resolve comments
* [Refactor] Refactor model inference pipeline (#52 )
* move attribute_to_dict to extract_model
* simplify the inference and visualization
* remove unused import
* [Feature] Support SRCNN in mmedit with ONNXRuntime and TensorRT (#45 )
* finish mmedit-ort
* edit __init__ files
* add noqa
* add tensorrt support
* 1. Rename "base.py"
2. Move srcnn.py to correct directory
* fix bugs
* remove figures
* align to refactor-v1
* update comment in srcnn
* fix lint
* newfunc -> new_func
* Add visualize.py
split visualize() in each codebase
* fix lint
* fix lint
* remove unnecessary code in ORTRestorer
* remove .api
* edit super(), remove dataset
* [Refactor]: Change name of split to partition (#57 )
* refactor mmcls configs
* refactor mmdet configs and split params
* rename rest split to partition from master
* remove base.py
* fix init of inference class
* fix mmocr init, add show_result alias
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: RunningLeon <maningsheng@sensetime.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
2021-09-01 15:43:49 +08:00
e5dc959276
[Feature]: Support PSPNet with static shape in ORT and TRT ( #39 )
...
* support pspnet with static shape
* set default as None to --test-img
2021-08-20 13:39:03 +08:00
fd40119694
[Feature] TensorRT int8 support ( #35 )
...
* fix custom ops support, fix multiple mark bug, add name mapping
* check if the value_info need to be added
* remove unnecessary print
* add nms implement
* two stage split wip
* add two stage split
* add split retinanet visualize
* add two stage split (wip)
* finish two stage split
* fix lint
* move parse string to mmdeploy.utils
* add func mark count dict
* use assert_cfg_valid
* update func count before add Mark
* fix dynamic shape support
* add calib data generator
* create calib dataset
* finish end2end int8
* add split two stage tensorrt visualize
2021-08-19 12:56:00 +08:00
7c465654b9
[Feature] Add dbnet18 and crnn support ( #36 )
...
* add mmocr
re-orgnize codes
save codes
save codes
save codes
add dbnet18
* fix lint
* support crnn
* fix lint
* resolve comments
* resolve conflicts
* resolve module registry error
* test
* resolve comments
* fix lint
* use algrithm_type and add dataset evaluation
* suit newest mmocr
* fix lint and remove trt version check
2021-08-19 11:53:00 +08:00
4c8c34d354
[Feature] Enable faster rcnn split onnxruntime ( #27 )
...
* fix custom ops support, fix multiple mark bug, add name mapping
* check if the value_info need to be added
* remove unnecessary print
* add nms implement
* two stage split wip
* add two stage split
* add split retinanet visualize
* add two stage split (wip)
* finish two stage split
* fix lint
* move parse string to mmdeploy.utils
* add func mark count dict
* use assert_cfg_valid
* update func count before add Mark
* fix dynamic shape support
2021-08-13 10:06:28 +08:00
e05521c933
[Feature] Merge NCNN deployment to grimoire based on mmcls - revert [ #25 ]( https://github.com/grimoire/deploy_prototype/pull/25 ) ( #30 )
...
* add
* change VulkanSDK to 1.2.176.1
* add ncnn cmakelist
* add ncnn source code as third party
* add all ncnn
* ncnn compile passed
* onnx2ncnn correctly
* fix code style
* merge_as_grimoire_design, only backend_ops, manually register.
* remove data and test sh
* remove build example
* remove config ncnn
* remove onnx2ncnn intermediate files
* remove other files auto-generated
* remove vulkan tools
* remove Vulkan, gitignore new rules, __init__ new lines
* rollback __init__ to grimoire
* remove pytorch version pending
* grimoire comments reply 1, 3, 4
* reply comment 5,6,7
* add auto definer, add python register
* fix lint
* add ncnn deploy support
* add model_wrapper, fix a typo bug, and add code comment for onnx2ncnn(WIP)
* add model wrapper ncnn
* fix lint
* fix pep8
* fix pre-commit-config.yaml paths
* fix import
* fix lint
* remove sys.path.append
* remove sys
* isort fix
* fix double quoted
* fix trailing space
* try fix isort
* fix clang-format-9
* fix requests
* fix all comments
* Fix typo
* test code for grimoire
* fix ops register
* new definere
* fix visualization of mmcls
* remove temp
* fix flake8
* fix seed-isort-config
* fix thirdparty
* fix thirdparty
* fix yapf
* fix third_party_sort
* fix third party
* fix clang-format
* try fix clang-format
* try to fix clang format 9 customreshape
* try fix clang-format-9
* try fix clang-format-9
* try fix clang-format-9
* try fix ext
* fix onnx2ncnn
* Fix comments
* Fix Comments
* Fix Comments
* Fix Comments
* Fix conflict
* Fix flake8
* Update .isort.cfg
* Update ncnn_ext.cpp
* Update ncnn_ext.cpp
* fix missing ncnn backend code
* delete out of date comments of gather.cpp
* add DeployBaseClassifier
* add return -100 error
* clear out-of-date to do comments
Co-authored-by: 韩睿 <SENSETIME\hanrui1@cn0614008774l.domain.sensetime.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: grimoire <streetyao@live.com>
2021-08-05 14:06:47 +08:00
f607f1965b
[Feature] Align datasets ( #29 )
...
* add test tool and re-orgnize apis.utils
* handle topk and refine codes
* add cls export and test support
* fix lint
* move ort into wrapper
* resolve conflicts
* resolve comments
* resolve conflicts
* resolve comments and padding mrcnn
* resolve comments
2021-08-03 17:12:44 +08:00
ff39377833
[Feature] Retinanet split support ( #19 )
...
* prepare for split
* mark multiclass nms
* redesign mark decorator to handle input/output names better (#17 )
* redesign mark decorator to handle input/output names better
* fixed a bug in rename_value; robustify extract_model
* fix lint
* fix log in deploy pipeline
* fix onnxruntime config
* fix lint
* update connect init
* rename is_inspecting
* simplify function_mark
Co-authored-by: Li Zhang <lzhang329@gmail.com>
2021-07-23 13:18:32 +08:00
9647ab12db
Add cls visualization ( #18 )
...
* add cls visualization tool
* fix lint
* remove reduandant
* align result
2021-07-22 09:58:53 +08:00
10a2385d01
Support exporting to ONNX for Faster R-CNN ( #20 )
...
* update rewriter for rpn head
* new nms implement
* update bbox_head to support fasterrcnn with tensorrt
* rewrite SingleRoIExtractor
* remove unnecessary import
* fix fcos and bbox_nms
* resolve comments
* resolve comments
Co-authored-by: grimoire <yaoqian@sensetime.com>
2021-07-21 19:46:23 +08:00
1718501b4f
[Feature] Support fsaf fcos ( #14 )
...
* add fcos and fsaf
* remove is_in_onnx_export
* fix lint
* fix lint
* remove forwards
2021-07-13 20:13:39 +08:00
AllentDan
Jelle Maas
tpoisonooo
Chen Xin
Li Zhang
OldDreamInWind
lvhan028
q.yao
hanrui1sensetime
RunningLeon
q.yao
Semyon Bevzyuk
Yifan Zhou
RunningLeon