03c95a1149
[Enhancement]: Optimize config utils ( #263 )
...
* Optimize config utils
* Update `get_backend`
* Add assert
2021-12-08 14:55:59 +08:00
f424fca509
[Doc] add mmocr performance benchmark ( #240 )
...
* add mmocr performance benchmark
* add notes
* fix typo
* add openvino for mmocr and use T4 results for dbnet18
2021-12-08 10:31:57 +08:00
4f70434070
[Fix] correct ncnn_DIR ( #261 )
...
* correct ncnn_DIR
* change NCNN_DIR to ncnn_DIR
2021-12-08 10:29:07 +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
ec1c15dc07
Fix load tensorrt plugin anyway ( #255 )
2021-12-07 21:06:32 +08:00
8f769e2355
Fix ncnn_ext build error ( #258 )
2021-12-07 17:00:39 +08:00
36124f6205
Merge sdk ( #251 )
...
* check in cmake
* move backend_ops to csrc/backend_ops
* check in preprocess, model, some codebase and their c-apis
* check in CMakeLists.txt
* check in parts of test_csrc
* commit everything else
* add readme
* update core's BUILD_INTERFACE directory
* skip codespell on third_party
* update trt_net and ort_net's CMakeLists
* ignore clion's build directory
* check in pybind11
* add onnx.proto. Remove MMDeploy's dependency on ncnn's source code
* export MMDeployTargets only when MMDEPLOY_BUILD_SDK is ON
* remove useless message
* target include directory is wrong
* change target name from mmdeploy_ppl_net to mmdeploy_pplnn_net
* skip install directory
* update project's cmake
* remove useless code
* set CMAKE_BUILD_TYPE to Release by force if it isn't set by user
* update custom ops CMakeLists
* pass object target's source lists
* fix lint end-of-file
* fix lint: trailing whitespace
* fix codespell hook
* remove bicubic_interpolate to csrc/backend_ops/
* set MMDEPLOY_BUILD_SDK OFF
* change custom ops build command
* add spdlog installation command
* update docs on how to checkout pybind11
* move bicubic_interpolate to backend_ops/tensorrt directory
* remove useless code
* correct cmake
* fix typo
* fix typo
* fix install directory
* correct sdk's readme
* set cub dir when cuda version < 11.0
* change directory where clang-format will apply to
* fix build command
* add .clang-format
* change clang-format style from google to file
* reformat csrc/backend_ops
* format sdk's code
* turn off clang-format for some files
* add -Xcompiler=-fno-gnu-unique
* fix trt topk initialize
* check in config for sdk demo
* update cmake script and csrc's readme
* correct config's path
* add cuda include directory, otherwise compile failed in case of tensorrt8.2
* clang-format onnx2ncnn.cpp
Co-authored-by: zhangli <lzhang329@gmail.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
2021-12-07 10:57:55 +08:00
eceeaf0159
[Docs] fix ncnn docs ( #249 )
...
* fix ncnn docs
* add prepare for protobuf
* move install protob position
2021-12-06 19:03:18 +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
2f6f6f87a1
Add get_mo_command. ( #247 )
2021-12-06 10:31:32 +08:00
cc72c00e61
[Fix]: Fix homonymic rewriters bugs ( #242 )
...
* Fix bug
* license
2021-12-03 18:31:58 +08:00
597350c07b
optimize markdown lint ( #245 )
2021-12-03 18:31:44 +08:00
38730041eb
[Fix]: input tensor not contiguous in backend wrappers. ( #250 )
...
* fix wrappers
* update naming
2021-12-03 13:47:22 +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
3b97f64385
Support YOLOX with TensorRT ( #237 )
...
* support trt
* fix test
2021-12-01 15:21:55 +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
f4b3db188e
Update CI ( #243 )
2021-12-01 11:05:38 +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
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
24c3134529
update ci ( #235 )
2021-11-29 14:41:23 +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
5c8479a45d
[Docs]: How to support new backends ( #222 )
...
* add doc
* Update
* Resolve comments
2021-11-25 19:59:19 +08:00
eab6d76838
Fix test ( #234 )
2021-11-25 19:41:31 +08:00
58993ccd63
[Fix]Fix utils test ( #232 )
...
* fix apis unit test
* remove tmp file
* delete test/test_mmedit and tests/test_mmseg
Please enter the commit message for your changes. Lines starting
* fix device to cpu
* fix tests/test_utils
* rename test_util
2021-11-25 19:41:21 +08:00
75fc0bde64
[Feature]add file to verify install and check env #issue#168 ( #199 )
...
* add file to verify install and env
* add openvino and ppl
* fix comment
* add __version__
* fix comment
* resolve comment
* fix lint
2021-11-25 19:05:07 +08:00
8a3e5e0464
【Fix】Fix apis unit test ( #231 )
...
* fix apis unit test
* remove tmp file
* delete test/test_mmedit and tests/test_mmseg
Please enter the commit message for your changes. Lines starting
* fix device to cpu
2021-11-25 18:48:52 +08:00
ad3cd5a352
[Docs] Trt speed docs ( #226 )
...
* add cls trt benchmark
* add benchmark.md
* fix ~
* resolve comments
* merge master
* caption Det
* resolve comments
* conditions -> settings
2021-11-25 18:48:40 +08:00
cab362a248
[Unittest] mmcls fix ut ( #228 )
...
* mmcls fix ut
* force add data folder
* remove debug info
* fix ut bugs
2021-11-25 18:48:11 +08:00
8c95bede4c
[Refactor] refactor mmocr unit tests ( #227 )
...
* refactor mmocr unit tests
* fix unit tests
* resolve comments
2021-11-25 18:11:24 +08:00
ed70afedcd
[Fix] Fix ops unit test ( #225 )
...
* fix tensorrt and onnxrt ops unit test
* fix ncnn unit test
2021-11-25 17:06:08 +08:00
64ee8db7b9
Fix bug ( #230 )
2021-11-25 17:05:54 +08:00
3755966359
fix ( #223 )
2021-11-25 15:51:38 +08:00
ca0014a971
[Doc]: Fix doc ( #219 )
...
* add reminder for pspnet in doc
* update table format
* fix format
* rename file
* fix comment
2021-11-25 14:51:21 +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
d742b428ee
[Docs] Add get_started.md ( #211 )
...
* first
* Add get_started.md
* update readme
* Apply suggestions from code review
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
* add how to find path
* fix for comment
* an -> a
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
2021-11-23 16:00:36 +08:00
7b369e0628
Update log format ( #203 )
2021-11-22 20:41:17 +08:00
417a2a0bf0
[Fix] fix empty image exporting bug ( #196 )
...
* fix empty image exporting bug
* solve black empty image
* concat zeros ahead
* add docstring
2021-11-18 13:59:41 +08:00
45a623966f
rm with-mask config and log .2f for timer.py ( #208 )
2021-11-17 17:31:26 +08:00
acf1dc5d88
[Enhancement] Update function name and docstring in mmdeploy.pytorch ( #191 )
...
* Update function name and docstring in mmdeploy.python
* remove in docstring
2021-11-17 14:20:29 +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
0043848a52
fix TRTBatchedNMS ( #204 )
...
Co-authored-by: Li Zhang <zhangli@sensetime.com>
2021-11-16 15:23:07 +08:00
49dd1cf678
[Enhancement]: Support VFNet from MMDetection for OpenVINO and ONNX Runtime. ( #195 )
...
* Add deform_conv_openvino.
* Add get_bboxes_of_vfnet_head.
* Fix vfnet and add test_get_bboxes_of_vfnet_head.
* Update docs.
* Fix test_shufflenetv2_backbone__forward for openvino.
* Fixes.
2021-11-16 10:59:59 +08:00
e142c72663
Fix post process ( #198 )
2021-11-15 16:00:44 +08:00
6f98c423a1
[Docstring]: Update docstring and names of rewrite in mmdet ( #194 )
...
* refactor mmdet docstring
* fix yapf
* fix lint
* fix docformatter
* fix docformatter --wrap-descriptions 79
* reply comments
* fix runningleon comments;
* fix misleading
2021-11-15 10:41:27 +08:00
fa626a58a0
[Docs]Fix docs about cfg ( #184 )
...
* add docs about config comment
* fix blank
* fix comment
* fix comment
* fix comment
* fix comment
2021-11-12 17:48:41 +08:00
1c86b3977f
[Fix]: Fix topk in post processing for ncnn ( #197 )
...
* Fix
* update
* Fix ncnn postprocessing
* recover code
2021-11-12 17:33:45 +08:00
c599ed5904
[Enhancement]: Support YOLOX from MMDetection for OpenVINO and ONNX Runtime. ( #190 )
...
* Add get_bboxes_of_yolox_head.
* Add test_yolox_head_get_bboxes.
* Update docs.
2021-11-12 10:39:42 +08:00
d4d415ee7e
Fix topk of trt_batched_nms ( #193 )
2021-11-11 19:19:44 +08:00
e2c9af0bb7
[Enhance]: update installation docs ( #189 )
...
* update doc
* resolve comments
2021-11-11 18:48:48 +08:00
Yifan Zhou
AllentDan
lvhan028
q.yao
hanrui1sensetime
RunningLeon
Semyon Bevzyuk
VVsssssk
lzhangzz