mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
4dd4d4851b
* 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
6 lines
214 B
Python
6 lines
214 B
Python
_base_ = ['./classification_static.py', '../_base_/backends/rknn.py']
|
|
|
|
onnx_config = dict(input_shape=[224, 224])
|
|
codebase_config = dict(model_type='end2end')
|
|
backend_config = dict(input_size_list=[[3, 224, 224]])
|