gmnj commited on
Commit
ffc0a36
1 Parent(s): 8b11eeb

Upload 3 files

Browse files

added fine-tuned model

Fine_Tuning_DetrObjectDetection_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb6f656dfc319ac726939358713a8b5622a12183d9c4d92a24246732196b26cd
3
+ size 166673570
config.json ADDED
@@ -0,0 +1,226 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_dropout": 0.0,
3
+ "activation_function": "relu",
4
+ "architectures": [
5
+ "DetrForObjectDetection"
6
+ ],
7
+ "attention_dropout": 0.0,
8
+ "auxiliary_loss": false,
9
+ "backbone": "resnet50",
10
+ "bbox_cost": 5,
11
+ "bbox_loss_coefficient": 5,
12
+ "class_cost": 1,
13
+ "classifier_dropout": 0.0,
14
+ "d_model": 256,
15
+ "decoder_attention_heads": 8,
16
+ "decoder_ffn_dim": 2048,
17
+ "decoder_layerdrop": 0.0,
18
+ "decoder_layers": 6,
19
+ "dice_loss_coefficient": 1,
20
+ "dilation": false,
21
+ "dropout": 0.1,
22
+ "encoder_attention_heads": 8,
23
+ "encoder_ffn_dim": 2048,
24
+ "encoder_layerdrop": 0.0,
25
+ "encoder_layers": 6,
26
+ "eos_coefficient": 0.1,
27
+ "giou_cost": 2,
28
+ "giou_loss_coefficient": 2,
29
+ "id2label": {
30
+ "0": "N/A",
31
+ "1": "person",
32
+ "10": "traffic light",
33
+ "11": "fire hydrant",
34
+ "12": "street sign",
35
+ "13": "stop sign",
36
+ "14": "parking meter",
37
+ "15": "bench",
38
+ "16": "bird",
39
+ "17": "cat",
40
+ "18": "dog",
41
+ "19": "horse",
42
+ "2": "bicycle",
43
+ "20": "sheep",
44
+ "21": "cow",
45
+ "22": "elephant",
46
+ "23": "bear",
47
+ "24": "zebra",
48
+ "25": "giraffe",
49
+ "26": "hat",
50
+ "27": "backpack",
51
+ "28": "umbrella",
52
+ "29": "shoe",
53
+ "3": "car",
54
+ "30": "eye glasses",
55
+ "31": "handbag",
56
+ "32": "tie",
57
+ "33": "suitcase",
58
+ "34": "frisbee",
59
+ "35": "skis",
60
+ "36": "snowboard",
61
+ "37": "sports ball",
62
+ "38": "kite",
63
+ "39": "baseball bat",
64
+ "4": "motorcycle",
65
+ "40": "baseball glove",
66
+ "41": "skateboard",
67
+ "42": "surfboard",
68
+ "43": "tennis racket",
69
+ "44": "bottle",
70
+ "45": "plate",
71
+ "46": "wine glass",
72
+ "47": "cup",
73
+ "48": "fork",
74
+ "49": "knife",
75
+ "5": "airplane",
76
+ "50": "spoon",
77
+ "51": "bowl",
78
+ "52": "banana",
79
+ "53": "apple",
80
+ "54": "sandwich",
81
+ "55": "orange",
82
+ "56": "broccoli",
83
+ "57": "carrot",
84
+ "58": "hot dog",
85
+ "59": "pizza",
86
+ "6": "bus",
87
+ "60": "donut",
88
+ "61": "cake",
89
+ "62": "chair",
90
+ "63": "couch",
91
+ "64": "potted plant",
92
+ "65": "bed",
93
+ "66": "mirror",
94
+ "67": "dining table",
95
+ "68": "window",
96
+ "69": "desk",
97
+ "7": "train",
98
+ "70": "toilet",
99
+ "71": "door",
100
+ "72": "tv",
101
+ "73": "laptop",
102
+ "74": "mouse",
103
+ "75": "remote",
104
+ "76": "keyboard",
105
+ "77": "cell phone",
106
+ "78": "microwave",
107
+ "79": "oven",
108
+ "8": "truck",
109
+ "80": "toaster",
110
+ "81": "sink",
111
+ "82": "refrigerator",
112
+ "83": "blender",
113
+ "84": "book",
114
+ "85": "clock",
115
+ "86": "vase",
116
+ "87": "scissors",
117
+ "88": "teddy bear",
118
+ "89": "hair drier",
119
+ "9": "boat",
120
+ "90": "toothbrush"
121
+ },
122
+ "init_std": 0.02,
123
+ "init_xavier_std": 1.0,
124
+ "is_encoder_decoder": true,
125
+ "label2id": {
126
+ "N/A": 0,
127
+ "airplane": 5,
128
+ "apple": 53,
129
+ "backpack": 27,
130
+ "banana": 52,
131
+ "baseball bat": 39,
132
+ "baseball glove": 40,
133
+ "bear": 23,
134
+ "bed": 65,
135
+ "bench": 15,
136
+ "bicycle": 2,
137
+ "bird": 16,
138
+ "blender": 83,
139
+ "boat": 9,
140
+ "book": 84,
141
+ "bottle": 44,
142
+ "bowl": 51,
143
+ "broccoli": 56,
144
+ "bus": 6,
145
+ "cake": 61,
146
+ "car": 3,
147
+ "carrot": 57,
148
+ "cat": 17,
149
+ "cell phone": 77,
150
+ "chair": 62,
151
+ "clock": 85,
152
+ "couch": 63,
153
+ "cow": 21,
154
+ "cup": 47,
155
+ "desk": 69,
156
+ "dining table": 67,
157
+ "dog": 18,
158
+ "donut": 60,
159
+ "door": 71,
160
+ "elephant": 22,
161
+ "eye glasses": 30,
162
+ "fire hydrant": 11,
163
+ "fork": 48,
164
+ "frisbee": 34,
165
+ "giraffe": 25,
166
+ "hair drier": 89,
167
+ "handbag": 31,
168
+ "hat": 26,
169
+ "horse": 19,
170
+ "hot dog": 58,
171
+ "keyboard": 76,
172
+ "kite": 38,
173
+ "knife": 49,
174
+ "laptop": 73,
175
+ "microwave": 78,
176
+ "mirror": 66,
177
+ "motorcycle": 4,
178
+ "mouse": 74,
179
+ "orange": 55,
180
+ "oven": 79,
181
+ "parking meter": 14,
182
+ "person": 1,
183
+ "pizza": 59,
184
+ "plate": 45,
185
+ "potted plant": 64,
186
+ "refrigerator": 82,
187
+ "remote": 75,
188
+ "sandwich": 54,
189
+ "scissors": 87,
190
+ "sheep": 20,
191
+ "shoe": 29,
192
+ "sink": 81,
193
+ "skateboard": 41,
194
+ "skis": 35,
195
+ "snowboard": 36,
196
+ "spoon": 50,
197
+ "sports ball": 37,
198
+ "stop sign": 13,
199
+ "street sign": 12,
200
+ "suitcase": 33,
201
+ "surfboard": 42,
202
+ "teddy bear": 88,
203
+ "tennis racket": 43,
204
+ "tie": 32,
205
+ "toaster": 80,
206
+ "toilet": 70,
207
+ "toothbrush": 90,
208
+ "traffic light": 10,
209
+ "train": 7,
210
+ "truck": 8,
211
+ "tv": 72,
212
+ "umbrella": 28,
213
+ "vase": 86,
214
+ "window": 68,
215
+ "wine glass": 46,
216
+ "zebra": 24
217
+ },
218
+ "mask_loss_coefficient": 1,
219
+ "max_position_embeddings": 1024,
220
+ "model_type": "detr",
221
+ "num_hidden_layers": 6,
222
+ "num_queries": 100,
223
+ "position_embedding_type": "sine",
224
+ "scale_embedding": false,
225
+ "transformers_version": "4.7.0.dev0"
226
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_resize": true,
4
+ "feature_extractor_type": "DetrFeatureExtractor",
5
+ "format": "coco_detection",
6
+ "image_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
+ ],
11
+ "image_std": [
12
+ 0.229,
13
+ 0.224,
14
+ 0.225
15
+ ],
16
+ "max_size": 1333,
17
+ "size": 800
18
+ }