mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmdeploy/configs/_base_/backends/rknn.py

7 lines
207 B
Python
Raw Normal View History

Add RKNN support. (#865) * save codes * support resnet and yolov3 * support yolox * fix lint * add mmseg support and a doc * add UT * update supported model list * fix ci * refine docstring * resolve comments * remote output_tensor_type * resolve comments * update readme
2022-09-06 11:48:39 +08:00
backend_config = dict(
type='rknn',
common_config=dict(
Add model conversion support to rv1126 (#1203) * WIP * fix interpolate * support yolov3 and retinanet * support seg * support ssd * supports both partition types for retinanet and ssd * mean std doc * update doc, add UT * support FSAF * rename configs * update dump info * update * python package installation doc * update doc * update doc * doc
2022-11-02 11:04:22 +08:00
target_platform='rv1126', # 'rk3588'
optimization_level=1),
Add rv1126 yolov3 support to sdk (#1280) * add yolov3 head to SDK * add yolov5 head to SDK * fix export-info and lint, add reverse check * fix lint * fix export info for yolo heads * add output_names to partition_config * fix typo * config * normalize config * fix * refactor config * fix lint and doc * c++ form * resolve comments * fix CI * fix CI * fix CI * float strides anchors * refine pipeline of rknn-int8 * config * rename func * refactor * rknn wrapper dict and fix typo * rknn wrapper output update, mmcls use end2end type * fix typo
2022-11-22 20:16:22 +08:00
quantization_config=dict(do_quantization=True, dataset=None))