RunningLeon
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
Yifan Zhou
cc72c00e61
[Fix]: Fix homonymic rewriters bugs ( #242 )
...
* Fix bug
* license
2021-12-03 18:31:58 +08:00
AllentDan
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
RunningLeon
3b97f64385
Support YOLOX with TensorRT ( #237 )
...
* support trt
* fix test
2021-12-01 15:21:55 +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
Yifan Zhou
eab6d76838
Fix test ( #234 )
2021-11-25 19:41:31 +08:00
VVsssssk
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
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
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
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
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
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
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
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
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
Yifan Zhou
227bebaade
[Enhancement]: Rename files in mmdeploy/core/rewriters ( #157 )
...
* Rename files
* lint
* update docstring
2021-10-27 15:12:00 +08:00
AllentDan
77131ea071
[Feature] Add exporting SARNet to onnxruntime backends. ( #134 )
...
* save codes
* save codes
* support sar exporting to onnxruntime and tensorrt
* use 4G max gpu space
* support sequential sar
* remove trt config for sar
* add sar unit tests
* add sar unit tests
* resolve conflicts
* remove torch.add rewrite in unit tests
2021-10-26 15:42:06 +08:00
Yifan Zhou
398bf2b386
[Refactor]: Refactor mmdeploy/core ( #140 )
...
* refactor function rewriter
* Pass unit test
* Fix test
* Fix test
* lint
* refactor function rewriter
* Pass unit test
* Fix test
* Fix test
* lint
* support openvino
* update
1. All type hints of cfg are set to Dict
2. Check if kwargs exist when calling patch_model
3. Complement tests
4. key name 'object' is set to '_object'
* lint
* Update docstring
* Fix tests
* Improve test
2021-10-26 10:30:14 +08:00
AllentDan
35b7341766
fix mmocr ( #147 )
2021-10-25 19:09:28 +08:00
Semyon Bevzyuk
ba7ae81d3e
[Unittest]: Test for CascadeRoIHead ( #141 )
...
* 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.
* Added test_cascade_roi_head.
* Fix onnxruntime outputs type.
2021-10-25 15:13:09 +08:00
hanrui1sensetime
9e227b228b
[UnitTest] mmocr unittest ( #130 )
...
* WIP test_mmocr 8 out of 20
* test_mmocr_export
* test mmocr apis
* add test data
* add mmocr model unittest 5 passed 1 failed
* finish mmocr unittest
* fix lint
* fix yapf
* fix isort
* fix flake8
* fix docformatter
* fix docformatter
* try to fix unittest after merge master
* Change test.py for backend.DEFAULT
* fix flake8
* fix ut
* fix yapf
* fix ut build
* fix yapf
* fix mmocr_export ut
* fix mmocr_apis ort not cuda
* remove explicit .forward
* remove backendwrapper
* simplify the crnn and dbnet config
* simplify instance_test.json
* add another case of decoder
* increase coverage of test_mmocr_models base_recognizer
* improve coverage
* improve encode_decoder coverage
* reply for grimoire codereview
* what if not check cuda?
* remove image data
* reply to runningleon code review
* fix fpnc
* fix lint
* try to fix CI UT error
* fix fpnc with and wo custom ops
* fix yapf
* skip fpnc when cuda is not ready in ci
* reply for code review
* reply for code review
* fix yapf
* reply for code review
* fix yapf
* fix conflict
* remove unmatched data path
* remove unnecessary comments
2021-10-25 10:15:57 +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
e240c1569f
[Unittest]: Test mmcls ( #135 )
...
* Add tests
* lint
* add data
* Remove redundant code
2021-10-19 20:30:51 +08:00
VVsssssk
fd75d57175
[Unittest]: Add MMSegmentation test ( #125 )
...
* 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 ncnn check
* fix problems
* fix problems
* fix dim problems
* resolve comments
* Fix SwitchBackendWrapper
* fix assert problems
* fix assert
* Remove comment
* merge master
Co-authored-by: SingleZombie <singlezombie@163.com>
2021-10-19 15:25:06 +08:00
Yifan Zhou
6f36717a72
[Unittest]: Test mmdeploy.pytorch ( #133 )
...
* add some tests
* Fix topk bug
* Fix bug in test
* Add remaining tests
* Remvoe debug code
* Delete redundant code
* Remove redundant code
2021-10-19 13:14:33 +08:00
VVsssssk
0f1cda0d3e
[Unittest]Add mmcv unit test ( #128 )
...
* 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 ncnn check
* fix problems
* fix problems
* add mmcv unit test
* add mmcv unit test
* modify scores
* fix format
* fix scores to rand
* remove blank line
Co-authored-by: SingleZombie <singlezombie@163.com>
2021-10-19 11:43:35 +08:00
Yifan Zhou
26cf2bdf05
[Enhancement]: Clarify the return value of get_rewrite_outputs ( #129 )
...
* Modify function and its call
* fix typo
2021-10-15 16:18:32 +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
hanrui1sensetime
07cb78bb7c
[Unittest] add ncnn unittest expand and fix expand.cpp bugs. ( #118 )
...
* add ncnn unittest expand
* rollback utils.py
* remove figures
2021-10-14 10:24:14 +08:00
AllentDan
d4828c7836
[Unittests] MMDet unittests ( #112 )
...
* add mmdet unittests
* remove redundant img meta info
* import wrapper from backend util
* force add wrapper
* use smaller nuance
* add seed everything
* add create input part and some inference part unittests
* fix lint
* skip ppl
* remove pyppl
* add dataset files
* import ncnn_ext inside ncnn warpper
* use cpu device to create input
* add pssd and ptsd unittests
* clear mmdet unittests
* wrap function to enable rewrite
* refine codes and resolve comments
* move mmdet inside test func
* remove redundant line
* test visualize in mmdeploy.apis
* use less memory
* resolve comments
* fix ci
* move pytest.skip inside test function and use 3 channel input
2021-10-13 17:24:11 +08:00
hanrui1sensetime
1ed6bc0fb5
refactor ncnn ops unittest ( #121 )
2021-10-12 14:24:56 +08:00
hanrui1sensetime
21f2b047a1
[Unittest] Test ncnn gather op and fix gather.cpp ( #114 )
...
* add shape constantofshape unittest for ncnn
* fix lint
* standarize import
* fix lint
* reply for code review
* reply for code review
* fix lint
* remove some hardcode
* fix lint
* reply for code review
* test gather and fix gather cpp code
* fix yapf
* fix clang-format
* reply for code review
* reply for code review
* fix lint
2021-10-09 14:06:49 +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
hanrui1sensetime
cba43e4c22
[Unittest] Add NCNN tensorslice unittest and fix tensorslice.cpp bugs. ( #115 )
...
* add tensorslice unittest
* reply code review
* fix lint
* fix typo
2021-10-09 11:42:32 +08:00
q.yao
f56a30025a
[Enhance] Add spell hook ( #116 )
...
* add spell hook
* fix code spell
2021-10-09 11:34:14 +08:00
q.yao
a335444a49
[Fix]: Remove TensorRT NMS plugin ( #113 )
...
* Remove TensorRT NMS plugin
* remove nms test
2021-10-09 11:14:38 +08:00
VVsssssk
43eee77fdd
[Unittest]Add ops test ( #108 )
...
* 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
* Update test_ops.py
2021-10-09 10:45:03 +08:00
hanrui1sensetime
6318e9fcf9
[Unittest] Add ncnn 'Shape' 'ConstantOfShape' unittest ( #107 )
...
* add shape constantofshape unittest for ncnn
* fix lint
* standarize import
* fix lint
* reply for code review
* reply for code review
* fix lint
* remove some hardcode
* fix lint
* reply for code review
2021-09-29 15:02:58 +08:00
AllentDan
5818095fe0
[Unittests] Add a demo for codebase rewrite part unittests ( #89 )
...
* save codes
* add test_model
* save codes
* wrap func
* reformat
* fix lint
* refine docstring
* remove pkl in .gitignore
* add pkl
* apply channel 3
* add function and trt backend rewrite unittest
* fix lint and typo
* add skip condition
* fix typo
* define deploy config inside func and keep ortwrapper original
* speed up and remove ctx
* only inference if no backends
* fix ci
* fix ci
* [Fix] Fix test_calibration (#101 )
* fix test calibration
* Modify cuda to cpu
* add tensorrt check
* Revert "[Fix] Fix test_calibration (#101 )"
This reverts commit 3f8b8384bfd880538050798d2567f1c137a36174.
Co-authored-by: maningsheng <mnsheng@yeah.net>
Co-authored-by: Yifan Zhou <singlezombie@163.com>
2021-09-28 19:29:58 +08:00
VVsssssk
0bef0513c6
[Unittest]: Add ncnn test exporter and topk test ( #84 )
...
* add ncnn test exporter in test_ops.py
* add ncnn test exporter in utils.py
* fix ncnn unittest
* fix lint
* fix lint
* fix lint isort
* remove ncnn roi_align pytest
* add ncnn topk unittest
* update to new api
* fix lint
* add comments
* skip class
* skip ncnn
Co-authored-by: hanrui1sensetime <hanrui1@sensetime.com>
Co-authored-by: maningsheng <mnsheng@yeah.net>
2021-09-28 14:20:04 +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
