narugo's picture
Upload dbnetpp_resnet50-oclip_fpnc_1200e_icdar2015/pipeline.json with huggingface_hub
c272287
raw
history blame
3.16 kB
{
"pipeline": {
"input": [
"img"
],
"output": [
"post_output"
],
"tasks": [
{
"type": "Task",
"module": "Transform",
"name": "Preprocess",
"input": [
"img"
],
"output": [
"prep_output"
],
"transforms": [
{
"color_type": "color_ignore_orientation",
"type": "LoadImageFromFile"
},
{
"keep_ratio": true,
"type": "Resize",
"size": [
4068,
1024
]
},
{
"type": "Normalize",
"to_rgb": true,
"mean": [
123.675,
116.28,
103.53
],
"std": [
58.395,
57.12,
57.375
]
},
{
"type": "Pad",
"size_divisor": 32
},
{
"type": "DefaultFormatBundle"
},
{
"meta_keys": [
"scale_factor",
"flip_direction",
"img_norm_cfg",
"valid_ratio",
"flip",
"pad_shape",
"img_shape",
"img_path",
"ori_shape",
"ori_filename",
"filename"
],
"type": "Collect",
"keys": [
"img"
]
}
]
},
{
"name": "dbnet",
"type": "Task",
"module": "Net",
"is_batched": true,
"input": [
"prep_output"
],
"output": [
"infer_output"
],
"input_map": {
"img": "input"
},
"output_map": {}
},
{
"type": "Task",
"module": "mmocr",
"name": "postprocess",
"component": "DBHead",
"params": {},
"output": [
"post_output"
],
"input": [
"prep_output",
"infer_output"
]
}
]
}
}