MrHaddad commited on
Commit
078a291
·
verified ·
1 Parent(s): c83ff4e

Upload DetrForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +8 -3
  2. model.safetensors +2 -2
config.json CHANGED
@@ -102,6 +102,7 @@
102
  "typical_p": 1.0,
103
  "use_bfloat16": false
104
  },
 
105
  "bbox_cost": 5,
106
  "bbox_loss_coefficient": 5,
107
  "class_cost": 1,
@@ -126,7 +127,9 @@
126
  "2": "LABEL_2",
127
  "3": "LABEL_3",
128
  "4": "LABEL_4",
129
- "5": "LABEL_5"
 
 
130
  },
131
  "init_std": 0.02,
132
  "init_xavier_std": 1.0,
@@ -137,7 +140,9 @@
137
  "LABEL_2": 2,
138
  "LABEL_3": 3,
139
  "LABEL_4": 4,
140
- "LABEL_5": 5
 
 
141
  },
142
  "mask_loss_coefficient": 1,
143
  "model_type": "detr",
@@ -147,7 +152,7 @@
147
  "position_embedding_type": "sine",
148
  "scale_embedding": false,
149
  "torch_dtype": "float32",
150
- "transformers_version": "4.35.2",
151
  "use_pretrained_backbone": null,
152
  "use_timm_backbone": false
153
  }
 
102
  "typical_p": 1.0,
103
  "use_bfloat16": false
104
  },
105
+ "backbone_kwargs": null,
106
  "bbox_cost": 5,
107
  "bbox_loss_coefficient": 5,
108
  "class_cost": 1,
 
127
  "2": "LABEL_2",
128
  "3": "LABEL_3",
129
  "4": "LABEL_4",
130
+ "5": "LABEL_5",
131
+ "6": "LABEL_6",
132
+ "7": "LABEL_7"
133
  },
134
  "init_std": 0.02,
135
  "init_xavier_std": 1.0,
 
140
  "LABEL_2": 2,
141
  "LABEL_3": 3,
142
  "LABEL_4": 4,
143
+ "LABEL_5": 5,
144
+ "LABEL_6": 6,
145
+ "LABEL_7": 7
146
  },
147
  "mask_loss_coefficient": 1,
148
  "model_type": "detr",
 
152
  "position_embedding_type": "sine",
153
  "scale_embedding": false,
154
  "torch_dtype": "float32",
155
+ "transformers_version": "4.38.2",
156
  "use_pretrained_backbone": null,
157
  "use_timm_backbone": false
158
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c6f0dd22d7acaaea73a46f1a60c5d9129d9e6ae1603e240c0dc2a9c523d89b5f
3
- size 166508868
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61b6d7b6033a248a6412fe639d2722f27b45e80358a51c0d5b00faa43fc6e83e
3
+ size 166510924