Commit 68661967 authored by limm's avatar limm
Browse files

add config module

parent 4353fa59
Pipeline #2808 canceled with stages
_base_ = ['./text-detection_static.py', '../../_base_/backends/ncnn.py']
onnx_config = dict(input_shape=None)
_base_ = [
'./text-detection_dynamic.py', '../../_base_/backends/onnxruntime-fp16.py'
]
_base_ = [
'./text-detection_dynamic.py', '../../_base_/backends/onnxruntime.py'
]
_base_ = ['./text-detection_static.py', '../../_base_/backends/onnxruntime.py']
onnx_config = dict(input_shape=None)
_base_ = ['./text-detection_dynamic.py', '../../_base_/backends/openvino.py']
onnx_config = dict(input_shape=None)
backend_config = dict(
model_inputs=[dict(opt_shapes=dict(input=[1, 3, 640, 640]))])
_base_ = ['./text-detection_dynamic.py', '../../_base_/backends/pplnn.py']
onnx_config = dict(input_shape=None)
backend_config = dict(model_inputs=dict(opt_shape=[1, 3, 640, 640]))
_base_ = ['./text-detection_dynamic.py', '../../_base_/backends/sdk.py']
codebase_config = dict(model_type='sdk')
backend_config = dict(pipeline=[
dict(type='LoadImageFromFile'),
dict(
type='LoadOCRAnnotations',
with_polygon=True,
with_bbox=True,
with_label=True),
dict(type='PackTextDetInputs', meta_keys=['img_path', 'ori_shape'])
])
_base_ = ['./text-detection_static.py', '../../_base_/backends/snpe.py']
onnx_config = dict(input_shape=None)
_base_ = ['../../_base_/onnx_config.py']
codebase_config = dict(type='mmocr', task='TextDetection')
_base_ = [
'./text-detection_dynamic.py', '../../_base_/backends/tensorrt-fp16.py'
]
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 320, 320],
opt_shape=[1, 3, 600, 800],
max_shape=[1, 3, 2240, 2240])))
])
_base_ = [
'./text-detection_static.py', '../../_base_/backends/tensorrt-fp16.py'
]
onnx_config = dict(input_shape=[512, 512])
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 512, 512],
opt_shape=[1, 3, 512, 512],
max_shape=[1, 3, 512, 512])))
])
_base_ = [
'./text-detection_dynamic.py', '../../_base_/backends/tensorrt-int8.py'
]
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 320, 320],
opt_shape=[1, 3, 600, 800],
max_shape=[1, 3, 2240, 2240])))
])
_base_ = [
'./text-detection_static.py', '../../_base_/backends/tensorrt-int8.py'
]
onnx_config = dict(input_shape=[512, 512])
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 512, 512],
opt_shape=[1, 3, 512, 512],
max_shape=[1, 3, 512, 512])))
])
_base_ = ['./text-detection_dynamic.py', '../../_base_/backends/tensorrt.py']
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 320, 320],
opt_shape=[1, 3, 600, 800],
max_shape=[1, 3, 2240, 2240])))
])
_base_ = ['./text-detection_static.py', '../../_base_/backends/tensorrt.py']
onnx_config = dict(input_shape=[512, 512])
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 512, 512],
opt_shape=[1, 3, 512, 512],
max_shape=[1, 3, 512, 512])))
])
_base_ = [
'../../_base_/torchscript_config.py',
'../../_base_/backends/torchscript.py'
]
ir_config = dict(input_shape=None)
codebase_config = dict(type='mmocr', task='TextDetection')
_base_ = ['./text-recognition_static.py']
onnx_config = dict(
dynamic_axes={
'input': {
0: 'batch',
3: 'width'
},
'output': {
0: 'batch',
1: 'seq_len',
2: 'num_classes'
}
}, )
_base_ = ['./text-recognition_static.py', '../../_base_/backends/ncnn.py']
onnx_config = dict(input_shape=None)
_base_ = [
'./text-recognition_dynamic.py',
'../../_base_/backends/onnxruntime-fp16.py'
]
_base_ = [
'./text-recognition_dynamic.py', '../../_base_/backends/onnxruntime.py'
]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment