Commit Graph

16 Commits (3eabeae071258335ec0963209e087fe7122a03be)

Author SHA1 Message Date
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
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
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
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
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 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
VVsssssk 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
Semyon Bevzyuk 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
hanrui1sensetime a35fb3ddd1
[Feature]: Support shufflenet_v2 for NCNN (#183)
* support shufflenet_v2

* fix yapf

* fix lint

* reply comments

* fix docformatter

* reply comment

* fix lint

* fix ut

* fix yapf
2021-11-10 20:14:31 +08:00
hanrui1sensetime 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
RunningLeon 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
Semyon Bevzyuk ca93007035
[Enhancement]: Support FoveaBox and ATSS from mmdetection for OpenVINO and ONNX Runtime (#160)
* Add FoveaHead and AtssHead.

* Update docs.

* Fix typos and docstrings.

* Fixes.

* Fix img_metas type in docstring.
2021-11-05 17:45:49 +08:00
Semyon Bevzyuk c52b24c67f
[Enhancement]: Added support for masks in OpenVINO. (#148)
* 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 support for masks (Mask RCNN).

* Add masks to CascadeRoIHead.simple_test.

* Added masks to test_OpenVINODetector.

* Added test_cascade_roi_head_with_mask.

* Update docs.

* Fix segm_results shape.

* Fix TopK in NMS and add test_multiclass_nms_with_keep_top_k.

* Removed unnecessary functions.

* Fix.

* Fix test_multiclass_nms_with_keep_top_k.

* Updated test_OpenVINODetector.
2021-11-03 10:27:48 +08:00
VVsssssk d3e26b68a2
fix docs bug (#166) 2021-11-02 19:12:53 +08:00
VVsssssk a357ffa924
[Docs]Add docs about How to convert model (#132)
* add ncnn test exporter in test_ops.py

* add ncnn test exporter in utils.py

* add onnxruntime and tensorrt ops test

* fix blank line

* fix comment
add nms ops test

* remove nms test

* add test sample
add dockerstring

* remove nms test

* fix grid_sample
add type hind

* fix problem

* fix dockerstring

* add nms batch_nms multi_level_roi_align

* add test data

* fix problem

* rm pkl file dependent

* rm file

* add docstring

* remove multi_level_dependce

* add mmseg module unittest

* add mmseg test

* add mmseg model unit test

* fix blankline

* rename file

* add syncbn2bn unit test

* add apis/export

* lint

* lint

* ??

* delete#

* fix problems

* add mmcv unit test

* add docs about how to convert model

* fix docs convert model

* add list

* rm mmcv test

* add mmocr model in list
fix decription

* fix docs

* fix docs

* refactor docs

* fix model list

* add docs about how to create config file

* fix how to convert model

* remove other docs

* add cn docs

* fix args description

* fix space

* fix about evaluate model

* fix link problems

* fix cn docs

Co-authored-by: SingleZombie <singlezombie@163.com>
2021-10-25 17:25:45 +08:00
q.yao 1d47688b32
[Docs]: Add sphinx document (#109)
* Add document structure

* better zh_cn document

* add build example, update requirement

* add readme

* add usage

* fix end of file

* fix codebase names, add TODO
2021-10-09 14:10:42 +08:00