zivpollak commited on
Commit
34f219a
1 Parent(s): ce56ac6

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": "zivpollak/detr-finetuned-iris-v1",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
6
- "DetrModel"
7
  ],
8
  "attention_dropout": 0.0,
9
  "auxiliary_loss": false,
@@ -122,7 +122,7 @@
122
  "giou_cost": 2,
123
  "giou_loss_coefficient": 2,
124
  "id2label": {
125
- "0": "iris"
126
  },
127
  "init_std": 0.02,
128
  "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,
 
122
  "giou_cost": 2,
123
  "giou_loss_coefficient": 2,
124
  "id2label": {
125
+ "0": "LABEL_0"
126
  },
127
  "init_std": 0.02,
128
  "init_xavier_std": 1.0,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:04648ac70786f09254527673ee6b768ee6baed0b8fed2099e44fe5c48455ef57
3
- size 165967248
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66368d45468d793a4a460107525ae2c8b4abcc63b254feb97609b3867d4f9a45
3
+ size 166503728