c5edb85550
Sync rv1126 to dev-1.x by cherry-pick ( #1295 )
...
* remove imports (#1207 )
* remove imports
* update doc
* detailed docstring
* rephrase
* Add model conversion support to rv1126 (#1203 )
* WIP
* fix interpolate
* support yolov3 and retinanet
* support seg
* support ssd
* supports both partition types for retinanet and ssd
* mean std doc
* update doc, add UT
* support FSAF
* rename configs
* update dump info
* update
* python package installation doc
* update doc
* update doc
* doc
* fix
* docstring
* remote partition config
2022-11-07 10:19:22 +08:00
4fc8828af8
fix ocr UT
...
fix ci and lint
fix det
fix cuda ci
fix mmdet test
update object detection
fix ut
fix layer norm ut
update ut
lock mmeit version
fix mmocr mmcls ut
add conftest.py
fix ocr ut
fix mmedit ci
install mmedit from source
fix rknn model and prepare_onnx_paddings__tensorrt UT
docstring
fix coreml export
update mmocr config
small test
recovery assert
fix ci
2022-09-29 16:37:36 +08:00
40adb329a3
Merge master:ea7706cb into sync_master
2022-09-22 19:49:50 +08:00
e3859004ba
add mod rewriter for TRT ( #1024 )
2022-09-21 14:47:36 +08:00
b0b502cba0
[Fix] Sync codebase 2.0 ( #979 )
...
* remove mmcv.Config
* fix mmdet with new mmengine
* fix ut
* fix mmdet ut
* fix sdk cpp pipeline
* fix mmcls
* fix mmcls ut
* adapt for new config
* fix cls models
* fix mmcls sdk
* fix yapf
* fix lint
* fix torchscript
* fix torchscript cpu
* fix check_env
* add test data
* fix export_info
* fix blank.jpg
* fix mmdet torchscript
* fix test.py
* fix sdk forward
* fix yapf
2022-09-15 21:54:57 +08:00
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
eed69b14e2
fix ci ( #964 )
2022-08-31 14:36:15 +08:00
f4decda86e
[Enhancement] Support DETR ( #924 )
...
* add detr support
* fix softmax
* add reg test, update document
2022-08-30 10:39:10 +08:00
18321eed51
[Bug] fix setitem with scalar or single element tensor ( #941 )
...
* fix setitem
* add copy symbolic
2022-08-29 18:56:17 +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
670a504502
fix adaptive_avg_pool exporting to onnx ( #857 )
...
* fix adaptive_avg_pool exporting to onnx
* remove debug codes
* fix ci
* resolve comment
2022-08-12 09:38:46 +08:00
36c35b6e88
[Fix] fix triu ( #792 )
...
* fix triu
* triu -> triu_default
2022-07-22 16:31:00 +08:00
0310c168dc
[Enhancement] Better index put ONNX export. ( #704 )
...
* Add rewriter for tensor setitem
* add version check
2022-07-20 19:32:46 +08:00
394fb55809
fix satrn for ORT ( #753 )
...
* fix satrn for ORT
* move rewrite into pytorch
2022-07-19 09:47:42 +08:00
da8c02e2b4
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
2022-06-27 17:52:05 +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
aa85760531
[Enhancement] Support litehrnet for ncnn ( #316 )
...
* support for litehrnet
* support ncnn vulkan
* update supported models
* add docstring
* add one ut and fix wrapper
* add test_shuffleunit
* add test_stem_forward
* add last ut
* fix flake8
* fix yapf
* fix lint
* fix yapf
* fix comments
* unified chunk
* mv adaptive_avg_pool
* fix lint
* move adaptive_avg_pool_ncnn
* fix lint
* symbolic rewriter of adaptive_avg_pool2d
* fix lint
* fix yapf
* fix ci
2022-04-29 16:46:19 +08:00
33bde90744
[Fix] Fix unittest of ncnn. ( #309 )
...
* fix test_pytorch_functions
* fix test_mmocr_models
2021-12-21 18:03:50 +08:00
fabdb473bb
Update ncnn test ( #298 )
...
* update ncnn test
* type hint
* update test ocr
* update mmseg ut
* ignore ncnn rpn head test
* add logging
* fix ssd base dense head test
* recover bacth in ncnn wrapper
* fix ncnn_ops_ut
* fix yapf
* recover test ops
* fix run_with_backend False
* Revert "fix run_with_backend False"
This reverts commit 83f8f915a25e800f5c2db339584d164ba40b2d9b.
* disable ncnn test test_pytorch_functions.py
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: hanrui1sensetime <hanrui1@sensetime.com>
2021-12-20 14:05:13 +08:00
0b23c744fc
remove trt in config ( #280 )
2021-12-13 19:57:15 +08:00
bb9b0a98de
[Enhancement]: Added static config and CI tests for OpenVINO. ( #218 )
...
* Add openvino_static.
* Add openvino-dev.
* Fix skipping ORT tests in test_mmocr_models.
* Updated docs.
* Fix print.
* Fix
* Fix
* Fix other backends
* Fix is_available
* fix ncnn
* Add constrict for get rewrite output
* add not
* Fix
* fix
* Fix
* Fix
* Improve tests
* Remove rebundant `cuda`
* Prevent None object and rename variable
* Fix multi-line string
* rename get_backend_checker
* Add Troubleshooting to doc.
* Fix postprocessing_masks with empty masks.
* Fix tests
* lint
* Update docs.
Co-authored-by: SingleZombie <singlezombie@163.com>
2021-12-01 14:03:48 +08:00
4fc0e79b4e
[Enhancement]: Refine backend environment check and fix online unit test ( #239 )
...
* Fix
* Fix other backends
* Fix is_available
* fix ncnn
* Add constrict for get rewrite output
* add not
* Fix
* fix
* Fix
* Fix
* Improve tests
* Remove rebundant `cuda`
* Prevent None object and rename variable
* Fix multi-line string
* rename get_backend_checker
* Fix tests
* lint
* docstring
2021-12-01 10:19:11 +08:00
a8abc75065
Add licence head to code ( #236 )
2021-11-30 15:00:37 +08:00
eab6d76838
Fix test ( #234 )
2021-11-25 19:41:31 +08:00
6f36717a72
[Unittest]: Test mmdeploy.pytorch ( #133 )
...
* add some tests
* Fix topk bug
* Fix bug in test
* Add remaining tests
* Remvoe debug code
* Delete redundant code
* Remove redundant code
2021-10-19 13:14:33 +08:00
AllentDan
AllentDan
RunningLeon
hanrui1sensetime
Li Zhang
q.yao
RunningLeon
Semyon Bevzyuk
Yifan Zhou
q.yao