ybzz commited on
Commit
c5987fa
1 Parent(s): f0301af

Upload DetrForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +3 -3
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
- "_name_or_path": "ybzz/detr-finetuned-pothole-v2",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
6
- "DetrModel"
7
  ],
8
  "attention_dropout": 0.0,
9
  "auxiliary_loss": false,
@@ -121,7 +121,7 @@
121
  "giou_cost": 2,
122
  "giou_loss_coefficient": 2,
123
  "id2label": {
124
- "0": "pothole"
125
  },
126
  "init_std": 0.02,
127
  "init_xavier_std": 1.0,
 
1
  {
2
+ "_name_or_path": "facebook/detr-resnet-50",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
6
+ "DetrForObjectDetection"
7
  ],
8
  "attention_dropout": 0.0,
9
  "auxiliary_loss": false,
 
121
  "giou_cost": 2,
122
  "giou_loss_coefficient": 2,
123
  "id2label": {
124
+ "0": "LABEL_0"
125
  },
126
  "init_std": 0.02,
127
  "init_xavier_std": 1.0,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:69d855f2ce467145140040f527a08f0bfcdac4808ebdd22a347b3ba89b683061
3
- size 165967248
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5151154a803c820bed283703e46a55af8a2abe7d717d05f9377bbaa32b905210
3
+ size 166503728