|
{
|
|
"version": "1.2.0",
|
|
"codebase": {
|
|
"task": "ObjectDetection",
|
|
"codebase": "mmdet",
|
|
"version": "3.1.0",
|
|
"pth": "rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth",
|
|
"config": "rtmdet_tiny_8xb32-300e_coco.py"
|
|
},
|
|
"codebase_config": {
|
|
"type": "mmdet",
|
|
"task": "ObjectDetection",
|
|
"model_type": "end2end",
|
|
"post_processing": {
|
|
"score_threshold": 0.05,
|
|
"confidence_threshold": 0.005,
|
|
"iou_threshold": 0.5,
|
|
"max_output_boxes_per_class": 200,
|
|
"pre_top_k": 5000,
|
|
"keep_top_k": 100,
|
|
"background_label_id": -1
|
|
}
|
|
},
|
|
"onnx_config": {
|
|
"type": "onnx",
|
|
"export_params": true,
|
|
"keep_initializers_as_inputs": false,
|
|
"opset_version": 11,
|
|
"save_file": "end2end.onnx",
|
|
"input_names": [
|
|
"input"
|
|
],
|
|
"output_names": [
|
|
"dets",
|
|
"labels"
|
|
],
|
|
"input_shape": null,
|
|
"optimize": true,
|
|
"dynamic_axes": {
|
|
"input": {
|
|
"0": "batch",
|
|
"2": "height",
|
|
"3": "width"
|
|
},
|
|
"dets": {
|
|
"0": "batch",
|
|
"1": "num_dets"
|
|
},
|
|
"labels": {
|
|
"0": "batch",
|
|
"1": "num_dets"
|
|
}
|
|
}
|
|
},
|
|
"backend_config": {
|
|
"type": "onnxruntime",
|
|
"precision": "fp16",
|
|
"common_config": {
|
|
"min_positive_val": 1e-07,
|
|
"max_finite_val": 10000.0,
|
|
"keep_io_types": false,
|
|
"disable_shape_infer": false,
|
|
"op_block_list": null,
|
|
"node_block_list": null
|
|
}
|
|
},
|
|
"calib_config": {}
|
|
} |