End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: SenseTime/deformable-detr
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: sensetime-deformable-detr-finetuned-10k-cppe5-manual-pad
@@ -14,31 +16,31 @@ should probably proofread and complete it, then remove this comment. -->
 [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/qubvel-hf-co/transformers-detection-model-finetuning-cppe5/runs/tqbxv4bu)
 # sensetime-deformable-detr-finetuned-10k-cppe5-manual-pad
 
-This model is a fine-tuned version of [SenseTime/deformable-detr](https://huggingface.co/SenseTime/deformable-detr) on an unknown dataset.
+This model is a fine-tuned version of [SenseTime/deformable-detr](https://huggingface.co/SenseTime/deformable-detr) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.0442
-- Map: 0.3514
-- Map 50: 0.6496
-- Map 75: 0.3279
-- Map Small: 0.2421
-- Map Medium: 0.2912
-- Map Large: 0.5328
-- Mar 1: 0.3162
-- Mar 10: 0.4852
-- Mar 100: 0.5091
-- Mar Small: 0.3118
-- Mar Medium: 0.4684
-- Mar Large: 0.7154
-- Map Coverall: 0.5877
-- Mar 100 Coverall: 0.6851
-- Map Face Shield: 0.3269
-- Mar 100 Face Shield: 0.4873
-- Map Gloves: 0.2902
-- Mar 100 Gloves: 0.4821
-- Map Goggles: 0.2064
-- Mar 100 Goggles: 0.4138
-- Map Mask: 0.3456
-- Mar 100 Mask: 0.4769
+- Loss: 1.0256
+- Map: 0.3589
+- Map 50: 0.6477
+- Map 75: 0.3379
+- Map Small: 0.2489
+- Map Medium: 0.3004
+- Map Large: 0.5406
+- Mar 1: 0.321
+- Mar 10: 0.4947
+- Mar 100: 0.5174
+- Mar Small: 0.3113
+- Mar Medium: 0.4731
+- Mar Large: 0.7397
+- Map Coverall: 0.5964
+- Mar 100 Coverall: 0.6878
+- Map Face Shield: 0.3426
+- Mar 100 Face Shield: 0.5165
+- Map Gloves: 0.2892
+- Mar 100 Gloves: 0.4808
+- Map Goggles: 0.2237
+- Mar 100 Goggles: 0.4215
+- Map Mask: 0.3424
+- Mar 100 Mask: 0.4804
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 99.53051643192488,
+    "eval_loss": 1.0255779027938843,
+    "eval_map": 0.3589,
+    "eval_map_50": 0.6477,
+    "eval_map_75": 0.3379,
+    "eval_map_Coverall": 0.5964,
+    "eval_map_Face_Shield": 0.3426,
+    "eval_map_Gloves": 0.2892,
+    "eval_map_Goggles": 0.2237,
+    "eval_map_Mask": 0.3424,
+    "eval_map_large": 0.5406,
+    "eval_map_medium": 0.3004,
+    "eval_map_small": 0.2489,
+    "eval_mar_1": 0.321,
+    "eval_mar_10": 0.4947,
+    "eval_mar_100": 0.5174,
+    "eval_mar_100_Coverall": 0.6878,
+    "eval_mar_100_Face_Shield": 0.5165,
+    "eval_mar_100_Gloves": 0.4808,
+    "eval_mar_100_Goggles": 0.4215,
+    "eval_mar_100_Mask": 0.4804,
+    "eval_mar_large": 0.7397,
+    "eval_mar_medium": 0.4731,
+    "eval_mar_small": 0.3113,
+    "eval_runtime": 15.6632,
+    "eval_samples_per_second": 9.577,
+    "eval_steps_per_second": 1.213,
+    "test_loss": 0.7814600467681885,
+    "test_map": 0.5279,
+    "test_map_50": 0.8341,
+    "test_map_75": 0.5627,
+    "test_map_Coverall": 0.6264,
+    "test_map_Face_Shield": 0.612,
+    "test_map_Gloves": 0.4778,
+    "test_map_Goggles": 0.4403,
+    "test_map_Mask": 0.4831,
+    "test_map_large": 0.705,
+    "test_map_medium": 0.4585,
+    "test_map_small": 0.5477,
+    "test_mar_1": 0.3727,
+    "test_mar_10": 0.621,
+    "test_mar_100": 0.6349,
+    "test_mar_100_Coverall": 0.7467,
+    "test_mar_100_Face_Shield": 0.7471,
+    "test_mar_100_Gloves": 0.5738,
+    "test_mar_100_Goggles": 0.5281,
+    "test_mar_100_Mask": 0.5788,
+    "test_mar_large": 0.8137,
+    "test_mar_medium": 0.5441,
+    "test_mar_small": 0.5739,
+    "test_runtime": 3.788,
+    "test_samples_per_second": 7.656,
+    "test_steps_per_second": 1.056,
+    "total_flos": 2.1870706694064443e+19,
+    "train_loss": 0.7253786786997093,
+    "train_runtime": 17217.4822,
+    "train_samples_per_second": 4.937,
+    "train_steps_per_second": 0.616
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 99.53051643192488,
+    "eval_loss": 1.0255779027938843,
+    "eval_map": 0.3589,
+    "eval_map_50": 0.6477,
+    "eval_map_75": 0.3379,
+    "eval_map_Coverall": 0.5964,
+    "eval_map_Face_Shield": 0.3426,
+    "eval_map_Gloves": 0.2892,
+    "eval_map_Goggles": 0.2237,
+    "eval_map_Mask": 0.3424,
+    "eval_map_large": 0.5406,
+    "eval_map_medium": 0.3004,
+    "eval_map_small": 0.2489,
+    "eval_mar_1": 0.321,
+    "eval_mar_10": 0.4947,
+    "eval_mar_100": 0.5174,
+    "eval_mar_100_Coverall": 0.6878,
+    "eval_mar_100_Face_Shield": 0.5165,
+    "eval_mar_100_Gloves": 0.4808,
+    "eval_mar_100_Goggles": 0.4215,
+    "eval_mar_100_Mask": 0.4804,
+    "eval_mar_large": 0.7397,
+    "eval_mar_medium": 0.4731,
+    "eval_mar_small": 0.3113,
+    "eval_runtime": 15.6632,
+    "eval_samples_per_second": 9.577,
+    "eval_steps_per_second": 1.213
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 99.53051643192488,
+    "test_loss": 0.7814600467681885,
+    "test_map": 0.5279,
+    "test_map_50": 0.8341,
+    "test_map_75": 0.5627,
+    "test_map_Coverall": 0.6264,
+    "test_map_Face_Shield": 0.612,
+    "test_map_Gloves": 0.4778,
+    "test_map_Goggles": 0.4403,
+    "test_map_Mask": 0.4831,
+    "test_map_large": 0.705,
+    "test_map_medium": 0.4585,
+    "test_map_small": 0.5477,
+    "test_mar_1": 0.3727,
+    "test_mar_10": 0.621,
+    "test_mar_100": 0.6349,
+    "test_mar_100_Coverall": 0.7467,
+    "test_mar_100_Face_Shield": 0.7471,
+    "test_mar_100_Gloves": 0.5738,
+    "test_mar_100_Goggles": 0.5281,
+    "test_mar_100_Mask": 0.5788,
+    "test_mar_large": 0.8137,
+    "test_mar_medium": 0.5441,
+    "test_mar_small": 0.5739,
+    "test_runtime": 3.788,
+    "test_samples_per_second": 7.656,
+    "test_steps_per_second": 1.056
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 99.53051643192488,
+    "total_flos": 2.1870706694064443e+19,
+    "train_loss": 0.7253786786997093,
+    "train_runtime": 17217.4822,
+    "train_samples_per_second": 4.937,
+    "train_steps_per_second": 0.616
+}