You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
3.6 KiB
121 lines
3.6 KiB
{
|
|
"pipeline": {
|
|
"input": [
|
|
"img"
|
|
],
|
|
"output": [
|
|
"post_output"
|
|
],
|
|
"tasks": [
|
|
{
|
|
"type": "Task",
|
|
"module": "Transform",
|
|
"name": "Preprocess",
|
|
"input": [
|
|
"img"
|
|
],
|
|
"output": [
|
|
"prep_output"
|
|
],
|
|
"transforms": [
|
|
{
|
|
"type": "LoadImageFromFile",
|
|
"backend_args": null
|
|
},
|
|
{
|
|
"type": "LetterResize",
|
|
"scale": [
|
|
640,
|
|
640
|
|
],
|
|
"allow_scale_up": false,
|
|
"pad_val": {
|
|
"img": 114
|
|
}
|
|
},
|
|
{
|
|
"type": "Normalize",
|
|
"to_rgb": true,
|
|
"mean": [
|
|
0.0,
|
|
0.0,
|
|
0.0
|
|
],
|
|
"std": [
|
|
255.0,
|
|
255.0,
|
|
255.0
|
|
]
|
|
},
|
|
{
|
|
"type": "Pad",
|
|
"size_divisor": 1
|
|
},
|
|
{
|
|
"type": "DefaultFormatBundle"
|
|
},
|
|
{
|
|
"type": "Collect",
|
|
"meta_keys": [
|
|
"filename",
|
|
"ori_shape",
|
|
"ori_filename",
|
|
"img_path",
|
|
"img_shape",
|
|
"flip_direction",
|
|
"flip",
|
|
"scale_factor",
|
|
"valid_ratio",
|
|
"pad_param",
|
|
"img_id",
|
|
"img_norm_cfg",
|
|
"pad_shape"
|
|
],
|
|
"keys": [
|
|
"img"
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "yolodetector",
|
|
"type": "Task",
|
|
"module": "Net",
|
|
"is_batched": false,
|
|
"input": [
|
|
"prep_output"
|
|
],
|
|
"output": [
|
|
"infer_output"
|
|
],
|
|
"input_map": {
|
|
"img": "input"
|
|
},
|
|
"output_map": {}
|
|
},
|
|
{
|
|
"type": "Task",
|
|
"module": "mmdet",
|
|
"name": "postprocess",
|
|
"component": "ResizeBBox",
|
|
"params": {
|
|
"multi_label": true,
|
|
"nms_pre": 30000,
|
|
"score_thr": 0.001,
|
|
"nms": {
|
|
"type": "nms",
|
|
"iou_threshold": 0.65
|
|
},
|
|
"max_per_img": 300
|
|
},
|
|
"output": [
|
|
"post_output"
|
|
],
|
|
"input": [
|
|
"prep_output",
|
|
"infer_output"
|
|
]
|
|
}
|
|
]
|
|
}
|
|
}
|