mmenendezg commited on
Commit
1f0202b
·
1 Parent(s): 65f57d9

Upload DetrForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +1 -9
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "facebook/detr-resnet-50",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
@@ -28,17 +28,9 @@
28
  "eos_coefficient": 0.1,
29
  "giou_cost": 2,
30
  "giou_loss_coefficient": 2,
31
- "id2label": {
32
- "0": "Negative",
33
- "1": "Positive"
34
- },
35
  "init_std": 0.02,
36
  "init_xavier_std": 1.0,
37
  "is_encoder_decoder": true,
38
- "label2id": {
39
- "Negative": 0,
40
- "Positive": 1
41
- },
42
  "mask_loss_coefficient": 1,
43
  "max_position_embeddings": 1024,
44
  "model_type": "detr",
 
1
  {
2
+ "_name_or_path": "../models/detr-resnet50-finetuned_brain_tumor",
3
  "activation_dropout": 0.0,
4
  "activation_function": "relu",
5
  "architectures": [
 
28
  "eos_coefficient": 0.1,
29
  "giou_cost": 2,
30
  "giou_loss_coefficient": 2,
 
 
 
 
31
  "init_std": 0.02,
32
  "init_xavier_std": 1.0,
33
  "is_encoder_decoder": true,
 
 
 
 
34
  "mask_loss_coefficient": 1,
35
  "max_position_embeddings": 1024,
36
  "model_type": "detr",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:090296487b79d9d3e20324be51a57ee0e16c161b5f3f63551f22013b73227045
3
  size 166608277
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4b58ebe53b600c580e40a4acacd9165966140bb0d502bccfc6039228298d7fb
3
  size 166608277