97e0d1228f
SDK ocr 2.0 ( #1006 )
...
* add deploy runner
* fix text_det wrapper
* fix recog
* save
* add crnn support
* update with_padding
* add short scale aspect jitter
* update regression test
* torch2ts
* add test data
* resolve comments
2022-09-19 15:08:51 +08:00
06028d6a21
Support mmseg:dev-1.x ( #790 )
...
* support pspnet + ort
* add rewriting for adapt_avg_pool
* test pspnet
* resize seg_pred to original image shape
* run with test.py
* keep as original
* fix ut of segmentation
* update var name
* fix export to torchscript
* sync with mmseg:test-1.x branch
* fix ut
* fix regression test for mmseg
* fix mmseg.ops
* update mmseg yml
* fix mmseg2.0 sdk
* fix adaptive pool
* update rewriting and tests
* fix sdk inputs
2022-09-14 20:08:52 +08:00
a2f82874bb
Support mmocr:dev-1.x ( #904 )
...
* init
* update UT
* fix UT except SAR
* update to latest 2.0
* fix ncnn UT
* export info
2022-09-01 15:11:43 +08:00
4d8ea40f55
Sync v0.7.0 to dev-1.x ( #907 )
...
* make -install -> make install (#621 )
change `make -install` to `make install`
https://github.com/open-mmlab/mmdeploy/issues/618
* [Fix] fix csharp api detector release result (#620 )
* fix csharp api detector release result
* fix wrong count arg of xxx_release_result in c# api
* [Enhancement] Support two-stage rotated detector TensorRT. (#530 )
* upload
* add fake_multiclass_nms_rotated
* delete unused code
* align with pytorch
* Update delta_midpointoffset_rbbox_coder.py
* add trt rotated roi align
* add index feature in nms
* not good
* fix index
* add ut
* add benchmark
* move to csrc/mmdeploy
* update unit test
Co-authored-by: zytx121 <592267829@qq.com>
* Reduce mmcls version dependency (#635 )
* fix shufflenetv2 with trt (#645 )
* fix shufflenetv2 and pspnet
* fix ci
* remove print
* ' -> " (#654 )
If there is a variable in the string, single quotes will ignored it, while double quotes will bring the variable into the string after parsing
* ' -> " (#655 )
same with https://github.com/open-mmlab/mmdeploy/pull/654
* Support deployment of Segmenter (#587 )
* support segmentor with ncnn
* update regression yml
* replace chunk with split to support ts
* update regression yml
* update docs
* fix segmenter ncnn inference failure brought by #477
* add test
* fix test for ncnn and trt
* fix lint
* export nn.linear to Gemm op in onnx for ncnn
* fix ci
* simplify `Expand` (#617 )
* Fix typo (#625 )
* Add make install in en docs
* Add make install in zh docs
* Fix typo
* Merge and add windows build
Co-authored-by: tripleMu <865626@163.com>
* [Enhancement] Fix ncnn unittest (#626 )
* optmize-csp-darknet
* replace floordiv to torch.div
* update csp_darknet default implement
* fix test
* [Enhancement] TensorRT Anchor generator plugin (#646 )
* custom trt anchor generator
* add ut
* add docstring, update doc
* Add partition doc and sample code (#599 )
* update torch2onnx tool to support onnx partition
* add model partition of yolov3
* add cn doc
* update torch2onnx tool to support onnx partition
* add model partition of yolov3
* add cn doc
* add to index.rst
* resolve comment
* resolve comments
* fix lint
* change caption level in docs
* update docs (#624 )
* Add java apis and demos (#563 )
* add java classifier detector
* add segmentor
* fix lint
* add ImageRestorer java apis and demo
* remove useless count parameter for Segmentor and Restorer, add PoseDetector
* add RotatedDetection java api and demo
* add Ocr java demo and apis
* remove mmrotate ncnn java api and demo
* fix lint
* sync java api folder after rebase to master
* fix include
* remove record
* fix java apis dir path in cmake
* add java demo readme
* fix lint mdformat
* add test javaapi ci
* fix lint
* fix flake8
* fix test javaapi ci
* refactor readme.md
* fix install opencv for ci
* fix install opencv : add permission
* add all codebases and mmcv install
* add torch
* install mmdeploy
* fix image path
* fix picture path
* fix import ncnn
* fix import ncnn
* add submodule of pybind
* fix pybind submodule
* change download to git clone for submodule
* fix ncnn dir
* fix README error
* simplify the github ci
* fix ci
* fix yapf
* add JNI as required
* fix Capitalize
* fix Capitalize
* fix copyright
* ignore .class changed
* add OpenJDK installation docs
* install target of javaapi
* simplify ci
* add jar
* fix ci
* fix ci
* fix test java command
* debugging what failed
* debugging what failed
* debugging what failed
* add java version info
* install openjdk
* add java env var
* fix export
* fix export
* fix export
* fix export
* fix picture path
* fix picture path
* fix file name
* fix file name
* fix README
* remove java_api strategy
* fix python version
* format task name
* move args position
* extract common utils code
* show image class result
* add detector result
* segmentation result format
* add ImageRestorer result
* add PoseDetection java result format
* fix ci
* stage ocr
* add visualize
* move utils
* fix lint
* fix ocr bugs
* fix ci demo
* fix java classpath for ci
* fix popd
* fix ocr demo text garbled
* fix ci
* fix ci
* fix ci
* fix path of utils ci
* update the circleci config file by adding workflows both for linux, windows and linux-gpu (#368 )
* update circleci by adding more workflows
* fix test workflow failure on windows platform
* fix docker exec command for SDK unittests
* Fixed tensorrt plugin not found in Windows (#672 )
* update introduction.png (#674 )
* [Enhancement] Add fuse select assign pass (#589 )
* Add fuse select assign pass
* move code to csrc
* add config flag
* remove bool cast
* fix export sdk info of input shape (#667 )
* Update get_started.md (#675 )
Fix backend model assignment
* Update get_started.md (#676 )
Fix backend model assignment
* [Fix] fix clang build (#677 )
* fix clang build
* fix ndk build
* fix ndk build
* switch to `std::filesystem` for clang-7 and later
* Deploy the Swin Transformer on TensorRT. (#652 )
* resolve conflicts
* update ut and docs
* fix ut
* refine docstring
* add comments and refine UT
* resolve comments
* resolve comments
* update doc
* add roll export
* check backend
* update regression test
* bump version to 0.6.0 (#680 )
* bump vertion to 0.6.0
* update version
* 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
* [Fix] fix ssd ncnn ut (#692 )
* fix ssd ncnn ut
* fix yapf
* fix passing img_metas to pytorch2onnx for mmedit (#700 )
* fix passing img_metas for mmdet3d (#707 )
* [Fix] Fix android build (#698 )
* fix android build
* fix cmake
* fix url link
* fix wrong exit code in pipeline_manager (#715 )
* fix exit
* change to general exit errorcode=1
* fix passing wrong backend type (#719 )
* 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
* [Fix] Fix mmdet ort static shape bug (#687 )
* fix shape
* add device
* fix yapf
* fix rewriter for transforms
* reverse image shape
* fix ut of distance2bbox
* fix rewriter name
* fix c4 for torchscript (#724 )
* [Enhancement] Standardize C API (#634 )
* unify C API naming
* fix demo and move apis/c/* -> apis/c/mmdeploy/*
* fix lint
* fix C# project
* fix Java API
* [Enhancement] Support Slide Vertex TRT (#650 )
* reorgnize mmrotate
* fix
* add hbb2obb
* add ut
* fix rotated nms
* update docs
* update benchmark
* update test
* remove ort regression test, remove comment
* Fix get-started rendering issues in readthedocs (#740 )
* fix mermaid markdown rendering issue in readthedocs
* fix error in C++ example
* fix error in c++ example in zh_cn get_started doc
* [Fix] set default topk for dump info (#702 )
* set default topk for dump info
* remove redundant docstrings
* add ci densenet
* fix classification warnings
* fix mmcls version
* fix logger.warnings
* add version control (#754 )
* fix satrn for ORT (#753 )
* fix satrn for ORT
* move rewrite into pytorch
* Add inference latency test tool (#665 )
* add profile tool
* remove print envs in profile tool
* set cudnn_benchmark to True
* add doc
* update tests
* fix typo
* support test with images from a directory
* update doc
* resolve comments
* [Enhancement] Add CSE ONNX pass (#647 )
* Add fuse select assign pass
* move code to csrc
* add config flag
* Add fuse select assign pass
* Add CSE for ONNX
* remove useless code
* Test robot
Just test robot
* Update README.md
Revert
* [Fix] fix yolox point_generator (#758 )
* fix yolox point_generator
* add a UT
* resolve comments
* fix comment lines
* limit markdown version (#773 )
* [Enhancement] Better index put ONNX export. (#704 )
* Add rewriter for tensor setitem
* add version check
* Upgrade Dockerfile to use TensorRT==8.2.4.2 (#706 )
* Upgrade TensorRT to 8.2.4.2
* upgrade pytorch&mmcv in CPU Dockerfile
* Delete redundant port example in Docker
* change 160x160-608x608 to 64x64-608x608 for yolov3
* [Fix] reduce log verbosity & improve error reporting (#755 )
* reduce log verbosity & improve error reporting
* improve error reporting
* [Enhancement] Support latest ppl.nn & ppl.cv (#564 )
* support latest ppl.nn
* fix pplnn for model convertor
* fix lint
* update memory policy
* import algo from buffer
* update ppl.cv
* use `ppl.cv==0.7.0`
* document supported ppl.nn version
* skip pplnn dependency when building shared libs
* [Fix][P0] Fix for torch1.12 (#751 )
* fix for torch1.12
* add comment
* fix check env (#785 )
* [Fix] fix cascade mask rcnn (#787 )
* fix cascade mask rcnn
* fix lint
* add regression
* [Feature] Support RoITransRoIHead (#713 )
* [Feature] Support RoITransRoIHead
* Add docs
* Add mmrotate models regression test
* Add a draft for test code
* change the argument name
* fix test code
* fix minor change for not class agnostic case
* fix sample for test code
* fix sample for test code
* Add mmrotate in requirements
* Revert "Add mmrotate in requirements"
This reverts commit 043490075e#792 )
* fix triu
* triu -> triu_default
* [Enhancement] Install Optimizer by setuptools (#690 )
* Add fuse select assign pass
* move code to csrc
* add config flag
* Add fuse select assign pass
* Add CSE for ONNX
* remove useless code
* Install optimizer by setup tools
* fix comment
* [Feature] support MMRotate model with le135 (#788 )
* support MMRotate model with le135
* cse before fuse select assign
* remove unused import
* [Fix] Support macOS build (#762 )
* fix macOS build
* fix missing
* add option to build & install examples (#822 )
* [Fix] Fix setup on non-linux-x64 (#811 )
* fix setup
* replace long to int64_t
* [Feature] support build single sdk library (#806 )
* build single lib for c api
* update csharp doc & project
* update test build
* fix test build
* fix
* update document for building android sdk (#817 )
Co-authored-by: dwSun <dwsunny@icloud.com>
* [Enhancement] support kwargs in SDK python bindings (#794 )
* support-kwargs
* make '__call__' as single image inference and add 'batch' API to deal with batch images inference
* fix linting error and typo
* fix lint
* improvement(sdk): add sdk code coverage (#808 )
* feat(doc): add CI
* CI(sdk): add sdk coverage
* style(test): code format
* fix(CI): update coverage.info path
* improvement(CI): use internal image
* improvement(CI): push coverage info once
* [Feature] Add C++ API for SDK (#831 )
* add C++ API
* unify result type & add examples
* minor fix
* install cxx API headers
* fix Mat, add more examples
* fix monolithic build & fix lint
* install examples correctly
* fix lint
* 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
* Support setup on environment with no PyTorch (#843 )
* support test with multi batch (#829 )
* support test with multi batch
* resolve comment
* import algorithm from buffer (#793 )
* [Enhancement] build sdk python api in standard-alone manner (#810 )
* build sdk python api in standard-alone manner
* enable MMDEPLOY_BUILD_SDK_MONOLITHIC and MMDEPLOY_BUILD_EXAMPLES in prebuild config
* link mmdeploy to python target when monolithic option is on
* checkin README to describe precompiled package build procedure
* use packaging.version.parse(python_version) instead of list(python_version)
* fix according to review results
* rebase master
* rollback cmake.in and apis/python/CMakeLists.txt
* reorganize files in install/example
* let cmake detect visual studio instead of specifying 2019
* rename whl name of precompiled package
* fix according to review results
* Fix SDK backend (#844 )
* fix mmpose python api (#852 )
* add prebuild package usage docs on windows (#816 )
* add prebuild package usage docs on windows
* fix lint
* update
* try fix lint
* add en docs
* update
* update
* udpate faq
* fix typo (#862 )
* [Enhancement] Improve get_started documents and bump version to 0.7.0 (#813 )
* simplify commands in get_started
* add installation commands for Windows
* fix typo
* limit markdown and sphinx_markdown_tables version
* adopt html <details open> tag
* bump mmdeploy version
* bump mmdeploy version
* update get_started
* update get_started
* use python3.8 instead of python3.7
* remove duplicate section
* resolve issue #856
* update according to review results
* add reference to prebuilt_package_windows.md
* fix error when build sdk demos
* fix mmcls
Co-authored-by: Ryan_Huang <44900829+DrRyanHuang@users.noreply.github.com>
Co-authored-by: Chen Xin <xinchen.tju@gmail.com>
Co-authored-by: q.yao <yaoqian@sensetime.com>
Co-authored-by: zytx121 <592267829@qq.com>
Co-authored-by: Li Zhang <lzhang329@gmail.com>
Co-authored-by: tripleMu <gpu@163.com>
Co-authored-by: tripleMu <865626@163.com>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: lvhan028 <lvhan_028@163.com>
Co-authored-by: Bryan Glen Suello <11388006+bgsuello@users.noreply.github.com>
Co-authored-by: zambranohally <63218980+zambranohally@users.noreply.github.com>
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: tpoisonooo <khj.application@aliyun.com>
Co-authored-by: Hakjin Lee <nijkah@gmail.com>
Co-authored-by: 孙德伟 <5899962+dwSun@users.noreply.github.com>
Co-authored-by: dwSun <dwsunny@icloud.com>
Co-authored-by: Chen Xin <irexyc@gmail.com>
2022-08-19 09:30:13 +08:00
0cb2f67b32
Fix mmpose regression test failed on mmpose v0.26.0 ( #576 )
...
* fix mmpose regression test
* try to fix lint
* fix lint
* remove old config
Co-authored-by: chenxin2 <chenxin2@sensetime.com>
2022-06-08 20:16:23 +08:00
ee878b539b
change resolution and fix regression test for mmocr ( #534 )
2022-06-02 18:29:41 +08:00
aabab46d8a
[Enhancement] Improve MMDeploy Regression test ( #425 )
...
* make regression test as a module under project
* using `--codebase` instead of `--deploy-yml`
* Improve doc for `--codebase`
* Add shorter arg `-p` for `--performance`
* make `checopoint-dir` as an arg for the script
* Gen error log when convert fail.
* Improve res code for testing
* Doc add sdk test detail
* Doc add env setup
* Fix lint
* Fix doc lint
* Improve model path in report
* Improve report title
* Improve report checkpoint path
* Fix lint
* move test yaml under `tests/regression`
* Improve the test yaml path
* Fix lint
* Improve doc
* make func `update_report` code better
* move doc to new location
* Fix arg
* Update arg details
* Use cpu when openvino and onnxruntime cpu package
* Fix word
* Fix func of openpyxl 3.0.9
* Add some info
* Fix lint
* Fix filename
* Fix doc link
* Fix dir name with space when is not sdk
* Add args `--models` for test specific model(s)
* not saving report when no model in codebase when using `--models`
* Fix doc
* Fix lint
* Add table for metric in doc
* Improve table for doc
* Using `None` install of `['all']`
* Improce doc
* set device type properly
* Increate popen bufsize
* Add `precision_type` in `work-dir`
* Fix popen stuck
* Fix lint
* Fix lint
* Fix popen stuck by using file handler
* Make metric dataset as a list
* Update mmseg.yml
* Remove 'FPS' in the report
* Update do_regression_test.md
* Improve log
* Fix codespell
* Fix doc
* ncnn only save `xxx.param` as checkpoint name in the report
Co-authored-by: maningsheng <mnsheng@yeah.net>
2022-05-27 17:08:32 +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
e3a8baac4c
[Enhancement] Support RotatedRetinaNet TensorRT ( #422 )
...
* add rotated nms trt plugin
* fix ops output shape
* rebase
* fix lint
* add fp16, benchmark
* format docs
* remove unused definition, add ut
* add docs
* update docs
* add doc
2022-05-24 10:34:22 +08:00
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
0ce7c83c63
mmrotate sdk module ( #450 )
...
* support mmrotate
* fix name
* windows default link to cudart_static.lib, which is not compatible with static build && python_api
* python api
* fix ci
* fix type & remove unused meta info
* fix doxygen, add [out] to @param
* fix mmrotate-c-api
* refactor naming
* refactor naming
* fix lint
* fix lint
* move replace_RResize -> get_preprocess
* Update cuda.cmake
On windows, make static lib and python api build success.
* fix ptr
* Use unique ptr to prevent memory leaks
* move unique_ptr
* remove deleter
Co-authored-by: chenxin2 <chenxin2@sensetime.com>
Co-authored-by: cx <cx@ubuntu20.04>
2022-05-17 23:37:32 +08:00
42dc5bc316
Support single stage rotated detector in MMRotate ( #428 )
...
* fix lint
* fix lint
* add mmrotate part
* update
* update
* fix
* remove init_detector
* success run with bs=1
* nms_rotated support batch
* support [batch_id, class_id, box_id]
* fix
* fix
* Create test_mmrotate_core.py
* add ut
* add ut
* Update nms_rotated.py
* fix
* Revert "fix"
This reverts commit f792387fb4
2022-05-07 16:11:43 +08:00
22652175bc
[Feature] Regression test for mmdeploy ( #302 )
...
* Add regression test script
* Add doc
* Add test yaml for mmdet
* evaluate_outputs Add return result
(cherry picked from commit c8c9cd75df1b8dcaa39e
2022-04-28 08:03:59 +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
6948e3d04b
Ncnn demo sdk ( #164 )
...
* fix different dims and data sync and fp16 macro
* fix pad is negative bug, but still need debugging info
* split detection_output to dets and labels
* fix clang++ compile bug
* for fp16 compile macro of cmake
* fix pad_val dict input
* fix yolox missing normalization and get int tensor
* about score_threshold
* remove -lstdc++fs for android
* move -pthread for android
* fix link libraries for CLANG++;
* fix clang-format
* for mobileyolov3 conf_thre
* fix lint
* fix bug
* support ncnn vulkan net
* remove unused debugging info;
* Change INFO to DEBUG
* support vulkan precision mapping
* fix pad
* optimize getdetslabels
* remove EVAL_MODE
* ncnn_net support 4-dim data.
* remove FP16 compile options
* remove use_gpu_vulkan compile options
* fix pad.cpp
* fix yapf
* fix clang-format
* rm redundant lines
* fix pad according to mmdet
* add android build docs
* fix lint
* use cpp style string comparision
* fix use after free bug
* Add missing -DBUILD_SHARED_LIBS=OFF for en
* Add missing -DBUILD_SHARED_LIBS=OFF for ZH-CN
* reset img2tensor_impl
* reset img2tensor_impl
* add blank line
* sync android.md docs
* fix some dirs
* fix docs
* update docs
* fix code
2022-03-25 23:28:16 +08:00
486d45e739
[FIX] update docs and configs about openvino ssd deployment ( #175 )
...
* add openvino static config for docs' ssd deploy
* fix docs of openvino.md
* fix ssd openvino deployment
* rename openvino config
* remove some files
2022-02-24 16:09:58 +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
0556feec79
Replace convert_syncbatchnorm in mmseg ( #93 )
...
* replace convert_syncbatchnorm with revert_sync_batchnorm from mmcv
* change logger
2022-01-26 11:07:28 +08:00
e6e32a9db4
[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
2022-01-19 13:54:45 +08:00
bb655af629
Add Python wrapper for SDK ( #27 )
...
* add python API for detector
* integrate detection
* add python segmentor
* add segmentation support
* add classifier, text-detector, text-recognizer and restorer
* integrate classifier
* integrate textdet, textrecog and restorer
* simplify
* add inst-seg
* fix inst-seg
* integrate inst-seg
* Moidfy _build_wrapper
* better pipeline substitution
* use registry for backend model creation
* build Python module according to C API targets
* minor fix
* move sdk data pipeline to backend_config
* remove debugging lines
* add docstring for SDKEnd2EndModel
* fix type hint
* fix lint
* fix lint
* insert build/lib to sys.path
Co-authored-by: SingleZombie <singlezombie@163.com>
2022-01-13 11:31:51 +08:00
76f96f09bc
remove partition configs and docs ( #59 )
2022-01-11 18:38:54 +08:00
9d4d52078b
【Fix】Fix ppl bug about grid sample ( #325 )
...
* fix ppl problems
* fix roialign
* fix grid_sampler bug
* fix grid sampler
* fix config
* fix test
2021-12-23 12:11:07 +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
f442dbbe83
[Docs]: MMCLS benchmark ( #300 )
...
* add doc
* Update benchmark
* Update benchmark
* Restore missing words
2021-12-17 14:12:26 +08:00
cde9abd52c
【Fix】Fix mmdet head ( #290 )
...
* fix anchor head and base dense head
* fix base dense head bug
* fix base dense head bug
* fix pad
* add ssd model int8 and fp16 config
* fix a bug about basedensehead
* add config for yolov3 trt fp16 int8
Co-authored-by: maningsheng <mnsheng@yeah.net>
2021-12-17 10:47:17 +08:00
3e8237d8bb
[Feature] Support end2end mmdet2.19 retina mobilessd ( #286 )
...
* support end2end mmdet2.19 retina mobilessd
* fix yapf
* add end2end fsaf
* fix lint
* fix comments
* fix lint
* add static configs
* fix docformatter
* move ssdhead
* add rewrite for l2norm
* fix ncnn ssd
* fix isort
* rename config
* add ssd_head_ut
* fix string
* align ssd
* remove unused bbox rewriter
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: maningsheng <mnsheng@yeah.net>
2021-12-17 10:46:54 +08:00
3eabeae071
Add new configs for ocr ( #295 )
...
* save configs
* add to benchmark
* add datasets to benchmark
* lock mmcor version to 0.3.0
2021-12-15 19:51:38 +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
0b23c744fc
remove trt in config ( #280 )
2021-12-13 19:57:15 +08:00
499d06d844
Add mmseg performance ( #254 )
...
* update doc and config
* test with ncnn
* update docs
* rename config
* update test result with openvino
* merge master
* resolve comments
* update dbnet
2021-12-09 20:17:00 +08:00
fa33b89261
[Docs]: Add MMEdit benchmark and supported model list ( #252 )
...
* doc
* update support
* update
* Resolve comments
* Update data
2021-12-09 11:27:20 +08:00
bca8cc0897
[fix] align the names of configs ( #256 )
...
* align names of configs
* set input_shape to None for all ppl
2021-12-07 21:06:44 +08:00
9e82851a73
[Feature] end2end yolov3 with ncnn ( #248 )
...
* support yolov3 ncnn with Yolov3DetectionOutput
* update nms
* fix contiguous in ncnn wrapper
* remove padding to detectionoutput
* format cpp
* Revert "format cpp"
This reverts commit 54050b19cd80d2f8cd851d82a755fd2c8d6c779d.
* fix zero detection
* fix yapf
* onnx2ncnn.cpp
* fix ut
* fix isort
* fix clang-format
* format cpp
* resolve comments
* resolve comments
* fix ut of ncnnend2endmodel
* fix yapf
* fix return list;
Co-authored-by: hanrui1sensetime <hanrui1@sensetime.com>
2021-12-06 19:01:36 +08:00
66d5cddbdc
[Enhancement] Add bicubic resize plugin for tensorrt ( #238 )
...
* save codes
* enable export fake bicubic interpolate op to onnx
* save codes
* enable bicubic interpolate trt plugin
* static export
* enable visualize but need align acc
* use torch bicubic upsample
* add unit tests for bicubic interpolate
* fix unit tests
* change mmedit config
* remove useless comments
* remove useless comments
* resolve comments
* fix lint
* clang-format
Co-authored-by: grimoire <yaoqian@sensetime.com>
2021-12-01 16:31:10 +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
cc2993e187
[Refactor] Refactor mmdet config ( #229 )
...
* Refactor mmdet config
* update partition model
* update docs related to mmdet
2021-11-30 15:00:11 +08:00
45a623966f
rm with-mask config and log .2f for timer.py ( #208 )
2021-11-17 17:31:26 +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
c05a839cfe
[FIX]fix with_mask to instance_seg ( #181 )
...
* fix with_mask to instance_seg
* fix instance_cfg to instance-cfg
2021-11-09 17:45:05 +08:00
778e3c9db0
[Enhance] Add SSD support for TensorRT ( #182 )
...
* support tensorrt ssd configs
* test ssd/cascade rcnn
* test shufflenetv1
* support trt mmdet end2end
2021-11-09 10:32:11 +08:00
cd51f12f32
[Feature]: Support yolov3 ( #167 )
...
* support yolov3 with ort and trt
* add ncnn compare
* fix yolo_head ncnn rewriter
* align perforance with ort trt for yolov3
* update doc
* add test for compare with equal,less, greater
* change namespace
* reformat cpp
* fix lint
* fix lint
* add unit test for yolov3 head
* remove compare op
* update doc
* update table format in docs
* update comments
* update
Co-authored-by: hanrui1sensetime <hanrui1@sensetime.com>
2021-11-08 16:07:58 +08:00
daee76ec99
[Enhancement] Optional export postprocess of mask ( #161 )
...
* Optional export postprocess of mask
* optional deploy_cfg
* delete print
2021-11-01 12:58:54 +08:00
5fdd75fba1
Add tensorrt int8 and fp16 configs for all the rest codebases ( #152 )
...
* add configs for all codebases except mmocr and fixed some bugs
* add ocr det
* add fp16 and int8 config for text recog
2021-10-27 15:59:02 +08:00
77131ea071
[Feature] Add exporting SARNet to onnxruntime backends. ( #134 )
...
* save codes
* save codes
* support sar exporting to onnxruntime and tensorrt
* use 4G max gpu space
* support sequential sar
* remove trt config for sar
* add sar unit tests
* add sar unit tests
* resolve conflicts
* remove torch.add rewrite in unit tests
2021-10-26 15:42:06 +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
5453f9befa
[Feature] Add tensorrt single stage partition ( #88 )
...
* Add tensorrt single stage partition
* add docstring&typehint
2021-09-23 15:49:36 +08:00
7e8154090a
align mmedit static cfg ( #82 )
2021-09-16 19:55:07 +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
3975d491d5
[Feature] Add ppl backend for ocr edit and seg ( #58 )
...
* add ppl for ocr edit and seg
* remove random crop after update mmedit
2021-09-07 15:40:43 +08:00
56bc1242b2
[Feature] Add NCNN on MMSegmentation ( #55 )
...
* 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 calib data generator
* create calib dataset
* finish end2end int8
* add split two stage tensorrt visualize
* first
* fix0
* fix1
* dirty work
* wip
* add allocator
* finally done!
* lint
* fix lint
* better gather
* better onnx2ncnn
* fix tensorslice bugs
* fix lint
* fix clang-format
* remove comments
* fix expand
* int param
* fix lint
* [Fix] NCNN TensorSlice op bugs (#42 )
* 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 calib data generator
* create calib dataset
* finish end2end int8
* add split two stage tensorrt visualize
* fix tensorslice bugs
* fix lint
* fix clang-format
* remove comments
* int param
* fix lint
Co-authored-by: grimoire <yaoqian@sensetime.com>
* add two stage ncnn support
* remove unused ops
* git unused config
* remove no_grad, should add in refactor
* add ncnn wrapper
* fix lint
* size return tuple
* Resolve grammar error
* Fix lint
* Trim Trailing Whitespace
* fix trim
* add argmax to topk
* add ArgMax parse
* add ncnn mmseg deploy cfg
* utils add ncnn mmseg
* add ncnn
* fix lint
* fix yapf
* fix clang-format-9
* remove debugging code
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: grimoire <streetyao@live.com>
Co-authored-by: maningsheng <mnsheng@yeah.net>
2021-09-03 15:19:54 +08:00
AllentDan
RunningLeon
Chen Xin
HinGwenWoong
tpoisonooo
q.yao
Yue Zhou
lvhan028
hanrui1sensetime
tripleMu
lzhangzz
VVsssssk
Yifan Zhou
RunningLeon
Semyon Bevzyuk