Commit Graph

781 Commits (c3986cebe8e3d5125e1b70fdde03bbbecfd0f6b1)
 

Author SHA1 Message Date
hanrui1sensetime 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
AllentDan 8c95bede4c
[Refactor] refactor mmocr unit tests (#227)
* refactor mmocr unit tests

* fix unit tests

* resolve comments
2021-11-25 18:11:24 +08:00
VVsssssk 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
Yifan Zhou 64ee8db7b9
Fix bug (#230) 2021-11-25 17:05:54 +08:00
RunningLeon 3755966359
fix (#223) 2021-11-25 15:51:38 +08:00
RunningLeon 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
q.yao 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
q.yao 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
q.yao 7b369e0628
Update log format (#203) 2021-11-22 20:41:17 +08:00
AllentDan 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
AllentDan 45a623966f
rm with-mask config and log .2f for timer.py (#208) 2021-11-17 17:31:26 +08:00
q.yao 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
AllentDan 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
lzhangzz 0043848a52
fix TRTBatchedNMS (#204)
Co-authored-by: Li Zhang <zhangli@sensetime.com>
2021-11-16 15:23:07 +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
Yifan Zhou e142c72663
Fix post process (#198) 2021-11-15 16:00:44 +08:00
hanrui1sensetime 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
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
Yifan Zhou 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
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
q.yao d4d415ee7e
Fix topk of trt_batched_nms (#193) 2021-11-11 19:19:44 +08:00
RunningLeon e2c9af0bb7
[Enhance]: update installation docs (#189)
* update doc

* resolve comments
2021-11-11 18:48:48 +08:00
RunningLeon b8e5ef00ea
[Refactor]: Update naming style and docstring in mmseg and mmcv (#188)
* update naming and docstring in mmseg and mmcv

* update docstring

* update self docstring

* resolve comments
2021-11-11 18:48:04 +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
VVsssssk c5ee3fc085
[Docstring]: Update mmcls and mmedit naming style and docstring (#187)
* fix mmcls and mmedit func name and docstring

* fix docstring

* fix class name

* fix comment

* add about self and ctx docstring
2021-11-10 19:44:00 +08:00
AllentDan e803430516
[Docstring] update docstrings and names of rewrite in mmocr (#180)
* update docstrings and names of rewrite in mmocr

* udpate names in __init__

* unit rewrite target description

* refine docstring

* fix typo

* fix lint

* remove useless arguments

* fix unit tests

* rm instance for self arg
2021-11-10 12:58:21 +08:00
VVsssssk c0fc3149d2
[Docs]How to add test units for backend ops (#146)
* 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

* ??

* add docs about how to add ops unit test

* fix diff

* fix docs

* fix docs

* remove cn docs

* fix comment

* fix comment

Co-authored-by: SingleZombie <singlezombie@163.com>
2021-11-10 10:53:19 +08:00
VVsssssk 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
q.yao cc4bfe15cb
fix bugs of build doc, update cmake version (#175) 2021-11-09 17:44:42 +08:00
q.yao 8208343afd
solve torch and cub conflict (#178) 2021-11-09 17:44:17 +08:00
q.yao 391ddeb9b5
Add licence info in backend_ops (#179) 2021-11-09 17:44:04 +08:00
AllentDan 79890f42c2
[Docs] add ppl install doc, how to test a model doc and mmocr doc (#169)
* add ppl install doc and how to test a model doc

* remove mmdet model list in ppl.md

* resolve comments

* fix typo

* add mmocr doc

* mv configs as the final column

* fix typo and align tables

* fix lint
2021-11-09 11:43:38 +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
Semyon Bevzyuk be995821f1
Fix postprocessing_masks. (#186) 2021-11-09 10:24:59 +08:00
del-zhenwu 80da064274
[Doc] Add openvino in doc (#176)
Signed-off-by: del-zhenwu <zhenxiang0708@163.com>

Co-authored-by: del-zhenwu <zhenxiang0708@163.com>
2021-11-08 16:09:18 +08:00
hanrui1sensetime ceae222a8a
[Docs] add mmseg.md (#174)
* add mmseg.md

* Align with pr #171

* left alignment of 1st and last column
2021-11-08 16:08:41 +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
RunningLeon 2ffc657665
[Docs]: Add mmcls support doc (#171)
* add doc

* update

* update

* update table format

* update name

* update

* resolve comments

* remove unnecessary files

* update mmedit

* update

* resolve comments
2021-11-08 15:37:43 +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
AllentDan 12eec39340
[Fix] fix trt_multi_level_roi_align plugin of tensorrt in T4 platform (#170)
* fix trt_multi_level_roi_align for tensorrt in T4 platform

* fix typo

* resolve comments
2021-11-04 15:53:48 +08:00
q.yao 4e168ee1c7
Update tensorrt install info (#165) 2021-11-03 10:55:50 +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
Yifan Zhou 6e4a32b746
[Doc]: How to add unit tests to rewritten models of codebases (#143)
* add english doc

* update index

* Resolve some comments

* Resolve comments

* Add Chinese doc

* Resolve comment
2021-11-02 10:33:20 +08:00
q.yao 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
q.yao ea3078aa56
Update custom ops namespace (#158)
* Update custom ops namespace

* rename register, rename generated library name

* rename domain to mmcv
2021-11-01 10:48:21 +08:00
VVsssssk 985bb6ad34
[Doc]How to write config (#139)
* 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 create config file

* fix :

* add zh docs about how to create config

* add full example

* fix comment

* add note

* fix problem

* fix catalog

* fix catalog`

* fix catalog

* fix docs

* fix cn docs

* fix lint

* fix docs

* fix space

* add mmocr link

* fix problem

* fix new

Co-authored-by: SingleZombie <singlezombie@163.com>
2021-10-29 18:04:11 +08:00
q.yao ef88d20241
lock version of mmdet (#159) 2021-10-29 10:59:16 +08:00
Yifan Zhou 31446de905
[Doc]: MMEdit doc (#153)
* Add doc

* Remove spaces

* sovle comments

* Resolve comments
2021-10-28 14:33:40 +08:00
AllentDan 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