RTMDet_PRODUCTION / model /detail.json
phuochungus's picture
fix broken git history
4986f6d
raw
history blame
1.89 kB
{
"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": {}
}