vision
image-detection
RTMDet-Detection / rtmdet_l_config.yaml
nijil-ti's picture
Add rtmdet model files
7a63d12 verified
Raw
History Blame Contribute Delete
2.13 kB
task_type: detection
dataloader:
name: coco_detection_dataloader
path: ./data/datasets/coco
preprocess:
resize: 640
crop: 640
data_layout: NCHW
reverse_channels: false
backend: cv2
interpolation: null
resize_with_pad:
- true
- corner
name: image_preprocess
pad_color:
- 114
- 114
- 114
session:
session_name: onnxrt
target_device: null
input_optimization: false
input_data_layout: NCHW
input_mean:
- 103.53
- 116.28
- 123.675
input_scale:
- 0.017429
- 0.017507
- 0.017125
model_path: rtmdet_l.onnx
model_id: od-mh8023
input_details: null
output_details: null
num_inputs: 1
postprocess:
reshape_list: null
formatter:
name: DetectionXYWH2XYXYCenterXY
resize_with_pad:
- true
- corner
normalized_detections: false
shuffle_indices: null
squeeze_axis: null
ignore_index: null
model_output_type: combined
concat_details:
axis: 2
end_index: 2
logits_bbox_to_bbox_ls:
score_fn: null
bbox_index: [0,4]
scores_index: [4,-1]
keypoint: false
object6dpose: false
name: detection_postprocess
metric:
label_offset_pred:
0: 1
1: 2
2: 3
3: 4
4: 5
5: 6
6: 7
7: 8
8: 9
9: 10
10: 11
11: 13
12: 14
13: 15
14: 16
15: 17
16: 18
17: 19
18: 20
19: 21
20: 22
21: 23
22: 24
23: 25
24: 27
25: 28
26: 31
27: 32
28: 33
29: 34
30: 35
31: 36
32: 37
33: 38
34: 39
35: 40
36: 41
37: 42
38: 43
39: 44
40: 46
41: 47
42: 48
43: 49
44: 50
45: 51
46: 52
47: 53
48: 54
49: 55
50: 56
51: 57
52: 58
53: 59
54: 60
55: 61
56: 62
57: 63
58: 64
59: 65
60: 67
61: 70
62: 72
63: 73
64: 74
65: 75
66: 76
67: 77
68: 78
69: 79
70: 80
71: 81
72: 82
73: 84
74: 85
75: 86
76: 87
77: 88
78: 89
79: 90
80: 91
model_info:
metric_reference:
accuracy_ap[.5:.95]%: 51.4
compact_name: rtmdet-l
shortlisted: true
recommended: false