RTMDet_PRODUCTION / model /pipeline.json
phuochungus's picture
fix broken git history
4986f6d
raw
history blame
4.11 kB
{
"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": "Resize",
"keep_ratio": true,
"size": [
640,
640
]
},
{
"type": "Pad",
"size": [
640,
640
],
"pad_val": {
"img": [
114,
114,
114
]
}
},
{
"type": "Normalize",
"to_rgb": false,
"mean": [
103.53,
116.28,
123.675
],
"std": [
57.375,
57.12,
58.395
]
},
{
"type": "Pad",
"size_divisor": 1
},
{
"type": "DefaultFormatBundle"
},
{
"type": "Collect",
"meta_keys": [
"img_path",
"flip",
"ori_filename",
"pad_param",
"pad_shape",
"scale_factor",
"valid_ratio",
"flip_direction",
"img_norm_cfg",
"img_id",
"img_shape",
"ori_shape",
"filename"
],
"keys": [
"img"
]
}
]
},
{
"name": "rtmdet",
"type": "Task",
"module": "Net",
"is_batched": true,
"input": [
"prep_output"
],
"output": [
"infer_output"
],
"input_map": {
"img": "input"
},
"output_map": {}
},
{
"type": "Task",
"module": "mmdet",
"name": "postprocess",
"component": "ResizeBBox",
"params": {
"nms_pre": 1000,
"min_bbox_size": 0,
"score_thr": 0.3,
"nms": {
"type": "nms",
"iou_threshold": 0.65
},
"max_per_img": 100
},
"output": [
"post_output"
],
"input": [
"prep_output",
"infer_output"
]
}
]
}
}