mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
124635ec5f
* 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
18 lines
609 B
Python
18 lines
609 B
Python
_base_ = ['../_base_/base_static.py', '../../_base_/backends/rknn.py']
|
|
|
|
onnx_config = dict(input_shape=[640, 640])
|
|
|
|
codebase_config = dict(model_type='rknn')
|
|
|
|
backend_config = dict(input_size_list=[[3, 640, 640]])
|
|
|
|
partition_config = dict(
|
|
type='rknn', # the partition policy name
|
|
apply_marks=True, # should always be set to True
|
|
partition_cfg=[
|
|
dict(
|
|
save_file='model.onnx', # name to save the partitioned onnx model
|
|
start=['detector_forward:input'], # [mark_name:input/output, ...]
|
|
end=['yolo_head:input']) # [mark_name:input/output, ...]
|
|
])
|