Commit
·
4ec38e0
1
Parent(s):
f30cbbc
Add model from convert_rt_detr_original_pytorch_checkpoint_to_pytorch.py
Browse files- 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,
|