RF-DETR-Detection / rfdetr_medium_config.yaml
mathmanu's picture
Add rfdetr model files
34566ff verified
Raw History Blame Contribute Delete
2.2 kB
task_type: detection
dataloader:
name: coco_detection_dataloader
path: ./data/datasets/coco
preprocess:
resize:
- 576
- 576
crop:
- 576
- 576
data_layout: NCHW
reverse_channels: false
backend: cv2
interpolation: null
resize_with_pad: false
pad_color:
- 0
- 0
- 0
name: image_preprocess
session:
input_optimization: false
input_data_layout: NCHW
input_mean:
- 123.675
- 116.28
- 103.53
input_scale:
- 0.017125
- 0.017507
- 0.017429
runtime_options: {}
model_path: rfdetr_medium.onnx
model_id: od-mh8042
input_details: null
output_details: null
num_inputs: 1
postprocess:
reshape_list: null
formatter:
name: DetectionCenterXYWH2XYXY
resize_with_pad: false
normalized_detections: true
shuffle_indices: null
squeeze_axis: null
ignore_index: null
model_output_type: split
logits_bbox_to_bbox_ls:
score_fn: sigmoid
bbox_index: 0
scores_index: 1
keypoint: false
object6dpose: false
name: detection_postprocess
metric:
label_offset_pred:
1: 1
2: 2
3: 3
4: 4
5: 5
6: 6
7: 7
8: 8
9: 9
10: 10
11: 11
12: 12
13: 13
14: 14
15: 15
16: 16
17: 17
18: 18
19: 19
20: 20
21: 21
22: 22
23: 23
24: 24
25: 25
26: 26
27: 27
28: 28
29: 29
30: 30
31: 31
32: 32
33: 33
34: 34
35: 35
36: 36
37: 37
38: 38
39: 39
40: 40
41: 41
42: 42
43: 43
44: 44
45: 45
46: 46
47: 47
48: 48
49: 49
50: 50
51: 51
52: 52
53: 53
54: 54
55: 55
56: 56
57: 57
58: 58
59: 59
60: 60
61: 61
62: 62
63: 63
64: 64
65: 65
66: 66
67: 67
68: 68
69: 69
70: 70
71: 71
72: 72
73: 73
74: 74
75: 75
76: 76
77: 77
78: 78
79: 79
80: 80
81: 81
82: 82
83: 83
84: 84
85: 85
86: 86
87: 87
88: 88
89: 89
90: 90
91: 91
0: 0
model_info:
metric_reference:
accuracy_ap[.5:.95]%: 54.7
model_shortlist: 10
compact_name: rfdetr-medium-576x576
shortlisted: true
recommended: false