Commit Graph

365 Commits (34879e638cc2db511e798a376b9a4b9932660fe1)
 

Author SHA1 Message Date
lvhan028 e10a1b7c9e
Add more demos (#305)
* add more examples

* change to restore's output image to BGR

* use BUILD_SHARED_LIBS

* update graph module's cmake

* add condition for spdlog package

* change cmakelists

* update cmakelists

* add device_name in each demo

* remove duplicate cmake function

* use palette to draw segmentation's result

* no need to find_package(pplnn) when use MMDeploy's static libs

* remove MMDeploy_LIBS' dependency on spdlog

* #include <opencv2/imgproc/imgproc.hpp>

* change from mmdeploy_core to mmdeploy::core
2021-12-21 21:32:39 +08:00
lzhangzz 56e32fdf63
improve shape checking (#315) 2021-12-21 20:16:40 +08:00
VVsssssk ce2b778061
[Fix]: fix pplnn empty output error (#320)
* fix ppl problems

* fix roialign
2021-12-21 19:49:06 +08:00
hanrui1sensetime 33bde90744
[Fix] Fix unittest of ncnn. (#309)
* fix test_pytorch_functions

* fix test_mmocr_models
2021-12-21 18:03:50 +08:00
Yifan Zhou c1ed41c465
[Docs] Refactor document structure (#310)
* Refactor docs

* Fix path

* Update logo
2021-12-21 13:59:07 +08:00
RunningLeon b060908a58
fix ut (#311) 2021-12-21 11:56:51 +08:00
lzhangzz 43e6714f47
[SDK] Fix asan errors (#308)
* fix several heap-use-after-free bugs

* rebase master

* add option for sanitizers

* rebase master

* fix conflicts
2021-12-21 10:47:21 +08:00
AllentDan 547a160770
Fix cub (#312)
* fix cub path

* use cmake source dir instead
2021-12-20 21:01:32 +08:00
RunningLeon d9b23bfc6f
[fix]: get tensor for DataContainer case (#307)
* fix DataContainer case for get_tensor_from_input

* fix docstring
2021-12-20 19:49:44 +08:00
Yifan Zhou 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
q.yao abdf64a576
[Fix] support ncnn faster-rcnn (#304)
* wtf

* Support fcos ncnn end2end

* support ncnn two stage detector

* fix test
2021-12-20 15:43:38 +08:00
RunningLeon 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
q.yao 270d98a8a2
[Fix] Support ncnn fcos end2end export (#303)
* wtf

* Support fcos ncnn end2end
2021-12-20 12:12:02 +08:00
RunningLeon ab5c51f3ab
fix mmdet tests (#302)
* fix mmdet tests

* fix
2021-12-18 14:36:54 +08:00
lvhan028 3be1779e66
Refactor tests (#283)
* fix sdk model's pipeline.json

* resize INT64 mask

* refactor unit tests

* fix api in model.h

* remove 'customs' from meta info

* fix zip model

* fix clang-format issue

* put tc on each backend into a SECTION

* change SECTION title

* add DYNAMIC_SECTION for capi unit test

* change 'devices' to 'device_names'

* change trt to tensorrt

* remove uncessary check

* add color_type 'color_ignore_orientation' which is used in ocr

* 'min_width', 'max_width' and 'backend' might be null in pipeline config

* fix clang-format issue

* remove useless code
2021-12-17 19:57:37 +08:00
AllentDan d3e8473f87
fix ci (#301)
* show mmcv

* test

* repalce

* refine
2021-12-17 14:30:11 +08:00
q.yao 75a9ffa611
fix ncnn linear (#299)
* fix ncnn linear

* lstm rewriter

* fix lint

* add comment
2021-12-17 14:12:37 +08:00
Yifan Zhou f442dbbe83
[Docs]: MMCLS benchmark (#300)
* add doc

* Update benchmark

* Update benchmark

* Restore missing words
2021-12-17 14:12:26 +08:00
q.yao e54d607f24
[SDK] Add OpenVINO SDK support (#257)
* Add OpenVINO SDK support

* fix reshape

* remove resize input
2021-12-17 13:45:32 +08:00
AllentDan f94d7b5abe
[Feature] Dump jsons to SDK (#279)
* save cods

* save codes

* save codes

* add docstring and fix ut

* use Resize to replace img_scale

* use Dict instead of dict()

* net && weights

* resolve comments

* update pipeline

* update pipeline

* update pipeline

* refactor preprocess

* merge master

* use imagetotensor

* use imagetotensor

* fix typos and improve docstring

* resolve comments

* update lq to img in mmedit

* add mask_thr_binary

* refactor dump info

* resolve comments

* add instance seg to dump info temperarily

* update meta_keys

* update meta_keys

* add meta_keys condition

* update postprocess for mmseg
2021-12-17 11:39:16 +08:00
VVsssssk 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
hanrui1sensetime 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
RunningLeon 31e8aed862
update ppl benchmark (#293) 2021-12-17 10:44:49 +08:00
lvhan028 1ed55f86f2
Update README (#288)
* update README and add three blank docs that are going to present SDK

* check in mmdeploy's logo image

* remove comments in README

* update acknowledgement

* development->deployment

* add Human3D link

* use captical word in citation
2021-12-16 19:22:14 +08:00
Yifan Zhou de6fc147bd
[Feature]: Add onnx2pplnn tool (#296)
* Add onnx2pplnn tool

* format

* typo

* update opt-shapes and wrapper

* Update opt-shape

* Update comment

* lint
2021-12-16 17:01:33 +08:00
lvhan028 f807346592
[SDK] update ppl.cv to latest version and use it's cmake package (#294) 2021-12-16 16:33:14 +08:00
lzhangzz a494a6f6ff
[SDK] sync changes according to performance benchmarks (#297)
* sync SDK changes according to performance benchmarks

* fix end-of-file lint

* fix clang-format issue

* fix clang-format by adding 'clang-format off'

* remove useless casts

* remove 'data' argument of 'operator()'

* change 'Tensor2Img' to 'TensorToImg' according to spec

* correct tensor's name according spec

Co-authored-by: lvhan028 <lvhan_028@163.com>
2021-12-16 13:51:22 +08:00
lvhan028 75b5939471
add ncnn custom ops to sdk (#284) 2021-12-15 21:34:32 +08:00
AllentDan 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
q.yao 43256977fa
Add onnx2ncnn tool (#276)
* Add onnx2ncnn tool

* fix docs

* fix lint
2021-12-15 18:43:09 +08:00
q.yao b1a202ef04
Add onnx2tensorrt tool (#275)
* add onnx2tensorrt

* add docs

* fix log

* fix docs
2021-12-15 11:20:10 +08:00
RunningLeon 469b6aab4c
Template for issue & pr (#287)
* add docs

* update

* update

* resolve comments
2021-12-14 16:30:44 +08:00
Yifan Zhou 78b37bbd32
[Fix]: Fix rewriter conflict when processing derived class (#289)
* Fix rewriter

* lint

* rename function and update docstring

* use is class

* Update docstring
2021-12-14 15:40:29 +08:00
AllentDan 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
RunningLeon 0b23c744fc
remove trt in config (#280) 2021-12-13 19:57:15 +08:00
AllentDan c8d0d6a285
fix dbnet export (#291) 2021-12-13 19:46:21 +08:00
RunningLeon aff568d508
[Fix]: update doc requirement (#274)
* update requirements

* fix version import
2021-12-13 19:36:30 +08:00
AllentDan 105acc9de9
[Fix] Dbnet performance of trt8 (#278)
* compatible trt version for dbnet

* judge inside rewrite
2021-12-13 16:44:05 +08:00
q.yao bd2867178d
Disable scatternd in TRT8+ (#277) 2021-12-13 16:43:22 +08:00
lvhan028 25a5838080
change trt to tensorrt and ort to onnxruntime according to sdk model spec (#282)
* change trt to tensorrt and ort to onnxruntime according to sdk model spec

* change ort to onnxruntime in config files
2021-12-13 14:58:01 +08:00
RunningLeon f2253a0f29
remove file (#285) 2021-12-13 13:01:30 +08:00
Semyon Bevzyuk a96e5f9d76
[Fix] Fix VFNet test (#281)
* [Fix] fix bugs for mmcls performance test (#269)

* fix bugs for mmcls performance test

* fix yapf

* add comments of CLASSES attribute

* Fix test_get_bboxes_of_vfnet_head

* Fix

Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
2021-12-13 10:43:39 +08:00
hanrui1sensetime 0f90a0af0a
[Fix] fix bugs for mmcls performance test (#269)
* fix bugs for mmcls performance test

* fix yapf

* add comments of CLASSES attribute
2021-12-10 14:15:08 +08:00
VVsssssk 54885e5972
【Feature】Support for mmdet2.19 (#253)
* add one-stage support

* add two-stage support

* fix docstring

* add base_dense_head docstring

* fix docstrign

* fix ncnn code

* fix anchor head

* fix unit test about mmdet model

* fix unit test

* fix base dense head bug:

* fix basehead

* fix basedense head

* fix pad bug

* fix bboxhead

* add rewrite about decode

* fix decode

* rename func

* fix test_mmdet_core and test_object_detection_model

* format test

* fix comments

* remove to(device)

* fix requirements

* fix docs

* fix yolox evaluate butg
2021-12-10 14:03:23 +08:00
Yifan Zhou 4149228716
[Enhancement]: Import codebase only when it is required (#266)
* Add import codebase

* lint

* Fix import order

* typo

* Fix partition

* docstring

* lint
2021-12-10 11:34:22 +08:00
RunningLeon 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
lvhan028 1948af6d06
eliminate inline template when build using release mode (#268) 2021-12-09 17:35:41 +08:00
q.yao 0897139744
[Fix] Fix onnx multiple domain registry (#270)
* Fix onnx multiple domain registry

* recover test args

* remove wrong status

* replace map with unordered_map

* add symbolic rewriter
2021-12-09 17:35:28 +08:00
AllentDan fd4297a2a3
add speed benchmark for pplnn and replace all ppl in docs with pplnn (#264) 2021-12-09 16:37:36 +08:00
Semyon Bevzyuk 3659b515eb
[Enhancement]: The ability to sort the dataset in tools/test.py. (#244)
* Add sort-data arg to test.py

* Set is_sort_dataset to True.

* Add a check for the possibility of sorting.

* lint

* Added mmdeploy.utils.dataset.

* Add unit test

Co-authored-by: SingleZombie <singlezombie@163.com>
2021-12-09 16:21:50 +08:00