rafaelpadilla commited on
Commit
4ec38e0
·
1 Parent(s): f30cbbc

Add model from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py

Browse files
Files changed (1) hide show
  1. config.json +18 -9
config.json CHANGED
@@ -1,5 +1,9 @@
1
  {
 
2
  "act_encoder": "silu",
 
 
 
3
  "backbone_config": {
4
  "backbone": "resnet50d",
5
  "freeze_batch_norm_2d": true,
@@ -41,6 +45,14 @@
41
  "id2label": {
42
  "0": "person",
43
  "1": "bicycle",
 
 
 
 
 
 
 
 
44
  "10": "fire hydrant",
45
  "11": "stop sign",
46
  "12": "parking meter",
@@ -51,7 +63,6 @@
51
  "17": "horse",
52
  "18": "sheep",
53
  "19": "cow",
54
- "2": "car",
55
  "20": "elephant",
56
  "21": "bear",
57
  "22": "zebra",
@@ -62,7 +73,6 @@
62
  "27": "tie",
63
  "28": "suitcase",
64
  "29": "frisbee",
65
- "3": "motorcycle",
66
  "30": "skis",
67
  "31": "snowboard",
68
  "32": "sports ball",
@@ -73,7 +83,6 @@
73
  "37": "surfboard",
74
  "38": "tennis racket",
75
  "39": "bottle",
76
- "4": "airplane",
77
  "40": "wine glass",
78
  "41": "cup",
79
  "42": "fork",
@@ -84,7 +93,6 @@
84
  "47": "apple",
85
  "48": "sandwich",
86
  "49": "orange",
87
- "5": "bus",
88
  "50": "broccoli",
89
  "51": "carrot",
90
  "52": "hot dog",
@@ -95,7 +103,6 @@
95
  "57": "couch",
96
  "58": "potted plant",
97
  "59": "bed",
98
- "6": "train",
99
  "60": "dining table",
100
  "61": "toilet",
101
  "62": "tv",
@@ -106,7 +113,6 @@
106
  "67": "cell phone",
107
  "68": "microwave",
108
  "69": "oven",
109
- "7": "truck",
110
  "70": "toaster",
111
  "71": "sink",
112
  "72": "refrigerator",
@@ -116,9 +122,7 @@
116
  "76": "scissors",
117
  "77": "teddy bear",
118
  "78": "hair drier",
119
- "79": "toothbrush",
120
- "8": "boat",
121
- "9": "traffic light"
122
  },
123
  "image_size": [
124
  640,
@@ -217,6 +221,7 @@
217
  "matcher_giou_cost": 2.0,
218
  "model_type": "rt_detr",
219
  "normalize_before": false,
 
220
  "num_decoder_layers": 6,
221
  "num_decoder_points": 4,
222
  "num_denoising": 100,
@@ -224,8 +229,12 @@
224
  "num_levels": 3,
225
  "num_queries": 300,
226
  "pe_temperature": 10000,
 
227
  "transformers_version": "4.36.0.dev0",
228
  "use_aux_loss": true,
 
 
 
229
  "use_focal_loss": true,
230
  "weight_loss_bbox": 5.0,
231
  "weight_loss_giou": 2.0,
 
1
  {
2
+ "_name_or_path": "rafaelpadilla/porting_rt_detr",
3
  "act_encoder": "silu",
4
+ "architectures": [
5
+ "RTDetrModel"
6
+ ],
7
  "backbone_config": {
8
  "backbone": "resnet50d",
9
  "freeze_batch_norm_2d": true,
 
45
  "id2label": {
46
  "0": "person",
47
  "1": "bicycle",
48
+ "2": "car",
49
+ "3": "motorcycle",
50
+ "4": "airplane",
51
+ "5": "bus",
52
+ "6": "train",
53
+ "7": "truck",
54
+ "8": "boat",
55
+ "9": "traffic light",
56
  "10": "fire hydrant",
57
  "11": "stop sign",
58
  "12": "parking meter",
 
63
  "17": "horse",
64
  "18": "sheep",
65
  "19": "cow",
 
66
  "20": "elephant",
67
  "21": "bear",
68
  "22": "zebra",
 
73
  "27": "tie",
74
  "28": "suitcase",
75
  "29": "frisbee",
 
76
  "30": "skis",
77
  "31": "snowboard",
78
  "32": "sports ball",
 
83
  "37": "surfboard",
84
  "38": "tennis racket",
85
  "39": "bottle",
 
86
  "40": "wine glass",
87
  "41": "cup",
88
  "42": "fork",
 
93
  "47": "apple",
94
  "48": "sandwich",
95
  "49": "orange",
 
96
  "50": "broccoli",
97
  "51": "carrot",
98
  "52": "hot dog",
 
103
  "57": "couch",
104
  "58": "potted plant",
105
  "59": "bed",
 
106
  "60": "dining table",
107
  "61": "toilet",
108
  "62": "tv",
 
113
  "67": "cell phone",
114
  "68": "microwave",
115
  "69": "oven",
 
116
  "70": "toaster",
117
  "71": "sink",
118
  "72": "refrigerator",
 
122
  "76": "scissors",
123
  "77": "teddy bear",
124
  "78": "hair drier",
125
+ "79": "toothbrush"
 
 
126
  },
127
  "image_size": [
128
  640,
 
221
  "matcher_giou_cost": 2.0,
222
  "model_type": "rt_detr",
223
  "normalize_before": false,
224
+ "num_classes": 80,
225
  "num_decoder_layers": 6,
226
  "num_decoder_points": 4,
227
  "num_denoising": 100,
 
229
  "num_levels": 3,
230
  "num_queries": 300,
231
  "pe_temperature": 10000,
232
+ "torch_dtype": "float32",
233
  "transformers_version": "4.36.0.dev0",
234
  "use_aux_loss": true,
235
+ "use_encoder_idx": [
236
+ 2
237
+ ],
238
  "use_focal_loss": true,
239
  "weight_loss_bbox": 5.0,
240
  "weight_loss_giou": 2.0,