mserrasa commited on
Commit
34a1db3
1 Parent(s): 5615196

End of training

Browse files
README.md ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: facebook/detr-resnet-50
4
+ tags:
5
+ - generated_from_trainer
6
+ model-index:
7
+ - name: detr-resnet-50_finetuned_cppe5
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # detr-resnet-50_finetuned_cppe5
15
+
16
+ This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the None dataset.
17
+
18
+ ## Model description
19
+
20
+ More information needed
21
+
22
+ ## Intended uses & limitations
23
+
24
+ More information needed
25
+
26
+ ## Training and evaluation data
27
+
28
+ More information needed
29
+
30
+ ## Training procedure
31
+
32
+ ### Training hyperparameters
33
+
34
+ The following hyperparameters were used during training:
35
+ - learning_rate: 1e-05
36
+ - train_batch_size: 8
37
+ - eval_batch_size: 8
38
+ - seed: 42
39
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
40
+ - lr_scheduler_type: linear
41
+ - num_epochs: 10
42
+ - mixed_precision_training: Native AMP
43
+
44
+ ### Training results
45
+
46
+
47
+
48
+ ### Framework versions
49
+
50
+ - Transformers 4.39.1
51
+ - Pytorch 2.2.1+cu121
52
+ - Datasets 2.18.0
53
+ - Tokenizers 0.15.2
config.json CHANGED
@@ -30,21 +30,41 @@
30
  "giou_cost": 2,
31
  "giou_loss_coefficient": 2,
32
  "id2label": {
33
- "0": "Coverall",
34
- "1": "Face_Shield",
35
- "2": "Gloves",
36
- "3": "Goggles",
37
- "4": "Mask"
 
 
 
 
 
 
 
 
 
 
38
  },
39
  "init_std": 0.02,
40
  "init_xavier_std": 1.0,
41
  "is_encoder_decoder": true,
42
  "label2id": {
43
- "Coverall": 0,
44
- "Face_Shield": 1,
45
- "Gloves": 2,
46
- "Goggles": 3,
47
- "Mask": 4
 
 
 
 
 
 
 
 
 
 
48
  },
49
  "mask_loss_coefficient": 1,
50
  "max_position_embeddings": 1024,
@@ -55,7 +75,7 @@
55
  "position_embedding_type": "sine",
56
  "scale_embedding": false,
57
  "torch_dtype": "float32",
58
- "transformers_version": "4.38.1",
59
  "use_pretrained_backbone": true,
60
  "use_timm_backbone": true
61
  }
 
30
  "giou_cost": 2,
31
  "giou_loss_coefficient": 2,
32
  "id2label": {
33
+ "0": "No finding",
34
+ "1": "Aortic enlargement",
35
+ "2": "Cardiomegaly",
36
+ "3": "Pleural thickening",
37
+ "4": "Pulmonary fibrosis",
38
+ "5": "Nodule/Mass",
39
+ "6": "Lung Opacity",
40
+ "7": "Pleural effusion",
41
+ "8": "Other lesion",
42
+ "9": "Infiltration",
43
+ "10": "ILD",
44
+ "11": "Calcification",
45
+ "12": "Consolidation",
46
+ "13": "Atelectasis",
47
+ "14": "Pneumothorax"
48
  },
49
  "init_std": 0.02,
50
  "init_xavier_std": 1.0,
51
  "is_encoder_decoder": true,
52
  "label2id": {
53
+ "Aortic enlargement": 1,
54
+ "Atelectasis": 13,
55
+ "Calcification": 11,
56
+ "Cardiomegaly": 2,
57
+ "Consolidation": 12,
58
+ "ILD": 10,
59
+ "Infiltration": 9,
60
+ "Lung Opacity": 6,
61
+ "No finding": 0,
62
+ "Nodule/Mass": 5,
63
+ "Other lesion": 8,
64
+ "Pleural effusion": 7,
65
+ "Pleural thickening": 3,
66
+ "Pneumothorax": 14,
67
+ "Pulmonary fibrosis": 4
68
  },
69
  "mask_loss_coefficient": 1,
70
  "max_position_embeddings": 1024,
 
75
  "position_embedding_type": "sine",
76
  "scale_embedding": false,
77
  "torch_dtype": "float32",
78
+ "transformers_version": "4.39.1",
79
  "use_pretrained_backbone": true,
80
  "use_timm_backbone": true
81
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fd086e7f2ffefa5c550eabe4b2147b59b19f086b78917404cb4d40426e3366a4
3
- size 166498936
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7389e7696686bb3a208f92b1dd167154368f00a708703a1434aab32e12fc90ec
3
+ size 166509224
preprocessor_config.json CHANGED
@@ -1,4 +1,24 @@
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "do_convert_annotations": true,
3
  "do_normalize": true,
4
  "do_pad": true,
 
1
  {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "annotations",
5
+ "return_segmentation_masks",
6
+ "masks_path",
7
+ "do_resize",
8
+ "size",
9
+ "resample",
10
+ "do_rescale",
11
+ "rescale_factor",
12
+ "do_normalize",
13
+ "do_convert_annotations",
14
+ "image_mean",
15
+ "image_std",
16
+ "do_pad",
17
+ "format",
18
+ "return_tensors",
19
+ "data_format",
20
+ "input_data_format"
21
+ ],
22
  "do_convert_annotations": true,
23
  "do_normalize": true,
24
  "do_pad": true,
runs/Mar26_09-53-28_6d27588d295c/events.out.tfevents.1711446808.6d27588d295c.2382.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f8bc6ec856dabaf1dad1895111a60ceddef68991ce0f583f88e40cbdc6816af
3
+ size 5957
runs/Mar26_09-53-28_6d27588d295c/events.out.tfevents.1711448631.6d27588d295c.2382.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b82566fa2091687b1b595981233f285668973fac5ff74dd50d1ec8432431062c
3
+ size 5957
runs/Mar26_09-53-28_6d27588d295c/events.out.tfevents.1711449032.6d27588d295c.2382.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03556ba205e23ccec7a1b56605078c9ff13b2e025f6386baaaa25c5ca5e544a1
3
+ size 5957
runs/Mar26_09-53-28_6d27588d295c/events.out.tfevents.1711451504.6d27588d295c.2382.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0cb53523acd7270a1c883d184d7953b3678d7b5dcfe39beff071facbf72b32f
3
+ size 5957
runs/Mar26_09-53-28_6d27588d295c/events.out.tfevents.1711451579.6d27588d295c.2382.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a5234b9a0170f13aaa5a288d7cd5fbc967a4911bc83d7e2f4c7b4c860b2eb3e
3
+ size 5957
runs/Mar26_09-53-28_6d27588d295c/events.out.tfevents.1711451701.6d27588d295c.2382.5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea8e27687ba10fa385881d767f805288e73301347decb16dfd666afa2acd6b7a
3
+ size 4184
runs/Mar26_09-53-28_6d27588d295c/events.out.tfevents.1711451734.6d27588d295c.2382.6 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:05ce4902bb90a06ac8b0f505a3d6e1146d121f5719b8795a8cc202c59774209d
3
+ size 6305
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c820a807cae84e0199e5f6a9370b48daa53927901915286c25aaa145164e7be0
3
  size 4920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:870ce42f9f2e0291757e9dd438c7064204c0cfaa5956fc64ddc10b3626ba1ae4
3
  size 4920