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
RunningLeon
0b23c744fc
remove trt in config ( #280 )
2021-12-13 19:57:15 +08:00
Semyon Bevzyuk
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
Yifan Zhou
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
q.yao
a8abc75065
Add licence head to code ( #236 )
2021-11-30 15:00:37 +08:00
AllentDan
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
VVsssssk
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
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
Semyon Bevzyuk
c5a87fb1bc
[Enhancement]: OpenVINO deployment support for some models ( #117 )
...
* 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 OpenVINODetector and onn2openvino tests.
* Add input_info to onnx2openvino.
* Add TestOpenVINOExporter and test_single_roi_extractor.
* Moved get_input_shape_from_cfg to openvino_utils.py and added test.
* Added test_cascade_roi_head.
* Add backend.check_env() to tests.
* Add OpenVINO to get_rewrite_outputs and to some tests in test_mmdet_models.
* Moved test_single_roi_extractor to test_mmdet_models.
* Removed TestOpenVINOExporter.
2021-10-21 16:07:35 +08:00
Yifan Zhou
10c4ef4203
[Unittest]: MMEdit unittest ( #124 )
...
* add mmedit test
* Solve unittest bug
* lint
* Remove debug code
* add data
* Refine code
* Fix ci
* Fix ci
* follow changes in mmdet test
* try to remove pytest skip
* remove redundant code
* rename
* Fix type hint
* Fix lint
* Refine SwitchBackendWrapper. Fix type hint
* update docstring
2021-10-15 10:25:48 +08:00
Yifan Zhou
f8a70f1d33
[Unittest]: Complement remaining tests in apis/core/utils ( #110 )
...
* Add remaining tests
* Use another method to import
* try to fix ppl import error
* try to fix pyppl import bug 2
* try to fix pyppl import error 3
* try to fix pyppl import error 4
* Use test class
* remove ()
* lint
* add else branch
2021-10-09 14:05:48 +08:00
Yifan Zhou
2073f3327e
[Fix] Fix test_calibration ( #101 )
...
* fix test calibration
* Modify cuda to cpu
2021-09-27 19:51:07 +08:00
RunningLeon
6ba1d45e81
[Fix]: fix ignoring test_ops ( #102 )
...
* show message from pytest skip
* skip utility class
Co-authored-by: SingleZombie <singlezombie@163.com>
2021-09-27 19:10:18 +08:00
RunningLeon
f9b2cab3fa
[Feature]: Update github CI ( #100 )
...
* align mmedit static cfg
* add for test
* update requirments
* add dependencies from mmlab
* change name
* lower thresh for interrogate at first
* update test
* update to skip
* Move import tensorrt
* Move import statement
Co-authored-by: SingleZombie <singlezombie@163.com>
2021-09-27 17:43:41 +08:00
Yifan Zhou
4de5bbb461
[Unittest]: Add demos for core/apis/utils ( #91 )
...
* add unittests
* lint
* modify .gitignore, remove useless files
* remove emtpy.py and generate it when use
* Update according to comments
1. Use tempfile
2. Delete inference test (which will be tested in each codebase)
3. Refine calibration test
* update annotation
* Add export_info
* Reduce data scale, fix assert
* update json blank line
* add backend check
2021-09-27 16:10:47 +08:00
q.yao
2b98040b26
[Feature] Refactor v1 ( #56 )
...
* [Refactor] add enum class and use functions to get configuration (#40 )
* add task and codebase enum class
* use funcitons to get config
* Refactor wrappers of mmcls and mmseg (#41 )
* move wrappers of cls & det to apis
* remove get_classes_from_config
* rename onnx_helper to onnx_utils
* move import to outside of class
* refactor ortwrappers
* Refactor build dataset and dataloader for mmseg (#44 )
* refactor build_dataset and build_dataloader for mmcls and mmseg
* remove repeated classes
* set build_dataloader with shuffle=False
* [Refactor] pplwrapper and mmocr refactor (#46 )
* add
* add pplwrapper and refactor mmocr
* fix lint
* remove unused arguments
* apply dict input for pplwrapper and ortwrapper
* add condition before import ppl and ort stuff
* update ppl (#51 )
* Refactor return value and extract_model (#54 )
* remove ret_value
* refactor extract_model
* fix typo
* resolve comments
* [Refactor] Refactor model inference pipeline (#52 )
* move attribute_to_dict to extract_model
* simplify the inference and visualization
* remove unused import
* [Feature] Support SRCNN in mmedit with ONNXRuntime and TensorRT (#45 )
* finish mmedit-ort
* edit __init__ files
* add noqa
* add tensorrt support
* 1. Rename "base.py"
2. Move srcnn.py to correct directory
* fix bugs
* remove figures
* align to refactor-v1
* update comment in srcnn
* fix lint
* newfunc -> new_func
* Add visualize.py
split visualize() in each codebase
* fix lint
* fix lint
* remove unnecessary code in ORTRestorer
* remove .api
* edit super(), remove dataset
* [Refactor]: Change name of split to partition (#57 )
* refactor mmcls configs
* refactor mmdet configs and split params
* rename rest split to partition from master
* remove base.py
* fix init of inference class
* fix mmocr init, add show_result alias
Co-authored-by: AllentDan <41138331+AllentDan@users.noreply.github.com>
Co-authored-by: RunningLeon <maningsheng@sensetime.com>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
2021-09-01 15:43:49 +08:00
RunningLeon
9f0e2d4339
changed codebase structure ( #10 )
...
* add delta2bbox rewriter
* rename onnx2trt
* add rewriter: anchor_generator_single_level_grid_priors
* Revert "add rewriter: anchor_generator_single_level_grid_priors"
This reverts commit ac7cf272942c4787bf143c0d67e414b0d2603b75.
* update comments
* remove clamp rewriter
* remove unused func
* reconstruct codes
* remomve mark
* create export package for codebases
* remove grid_sampler symbolic
* reorganize nms parts
* fix conflicts
* update unittest
2021-07-12 16:26:44 +08:00
q.yao
66a099faf9
[Feature] Apis unit test ( #7 )
...
* add apis test
* split torch2onnx impl, prepare for codebase test
* add is_available to backend
* lint
2021-07-05 12:51:43 +08:00
