mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
c5edb85550
* remove imports (#1207) * remove imports * update doc * detailed docstring * rephrase * 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 * fix * docstring * remote partition config
30 lines
924 B
Python
30 lines
924 B
Python
_base_ = ['../_base_/base_static.py', '../../_base_/backends/rknn.py']
|
|
|
|
onnx_config = dict(input_shape=[320, 320])
|
|
|
|
codebase_config = dict(model_type='rknn')
|
|
|
|
backend_config = dict(input_size_list=[[3, 320, 320]])
|
|
|
|
# yolov3, yolox
|
|
# 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
|
|
# start=['detector_forward:input'], # [mark_name:input, ...]
|
|
# end=['yolo_head:input']) # [mark_name:output, ...]
|
|
# ])
|
|
|
|
# # retinanet, ssd, fsaf
|
|
# partition_config = dict(
|
|
# type='rknn', # the partition policy name
|
|
# apply_marks=True,
|
|
# partition_cfg=[
|
|
# dict(
|
|
# save_file='model.onnx',
|
|
# start='detector_forward:input',
|
|
# end=['BaseDenseHead:output'])
|
|
# ])
|