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-auto-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/b2zl20b3)
 # sensetime-deformable-detr-finetuned-10k-cppe5-auto-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.1077
-- Map: 0.2966
-- Map 50: 0.5533
-- Map 75: 0.2796
-- Map Small: 0.1574
-- Map Medium: 0.2376
-- Map Large: 0.4613
-- Mar 1: 0.2796
-- Mar 10: 0.4669
-- Mar 100: 0.5019
-- Mar Small: 0.2991
-- Mar Medium: 0.4538
-- Mar Large: 0.701
-- Map Coverall: 0.542
-- Mar 100 Coverall: 0.673
-- Map Face Shield: 0.2897
-- Mar 100 Face Shield: 0.5063
-- Map Gloves: 0.1962
-- Mar 100 Gloves: 0.4442
-- Map Goggles: 0.1636
-- Mar 100 Goggles: 0.4262
-- Map Mask: 0.2918
-- Mar 100 Mask: 0.46
+- Loss: 1.1019
+- Map: 0.298
+- Map 50: 0.5545
+- Map 75: 0.2748
+- Map Small: 0.1608
+- Map Medium: 0.2403
+- Map Large: 0.4637
+- Mar 1: 0.2807
+- Mar 10: 0.469
+- Mar 100: 0.5006
+- Mar Small: 0.3003
+- Mar Medium: 0.4532
+- Mar Large: 0.6962
+- Map Coverall: 0.5405
+- Mar 100 Coverall: 0.6698
+- Map Face Shield: 0.2887
+- Mar 100 Face Shield: 0.5089
+- Map Gloves: 0.2012
+- Mar 100 Gloves: 0.4455
+- Map Goggles: 0.1644
+- Mar 100 Goggles: 0.4185
+- Map Mask: 0.2954
+- Mar 100 Mask: 0.4604
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 99.53051643192488,
+    "eval_loss": 1.101932406425476,
+    "eval_map": 0.298,
+    "eval_map_50": 0.5545,
+    "eval_map_75": 0.2748,
+    "eval_map_Coverall": 0.5405,
+    "eval_map_Face_Shield": 0.2887,
+    "eval_map_Gloves": 0.2012,
+    "eval_map_Goggles": 0.1644,
+    "eval_map_Mask": 0.2954,
+    "eval_map_large": 0.4637,
+    "eval_map_medium": 0.2403,
+    "eval_map_small": 0.1608,
+    "eval_mar_1": 0.2807,
+    "eval_mar_10": 0.469,
+    "eval_mar_100": 0.5006,
+    "eval_mar_100_Coverall": 0.6698,
+    "eval_mar_100_Face_Shield": 0.5089,
+    "eval_mar_100_Gloves": 0.4455,
+    "eval_mar_100_Goggles": 0.4185,
+    "eval_mar_100_Mask": 0.4604,
+    "eval_mar_large": 0.6962,
+    "eval_mar_medium": 0.4532,
+    "eval_mar_small": 0.3003,
+    "eval_runtime": 15.523,
+    "eval_samples_per_second": 9.663,
+    "eval_steps_per_second": 1.224,
+    "test_loss": 0.9166296720504761,
+    "test_map": 0.4366,
+    "test_map_50": 0.7304,
+    "test_map_75": 0.4448,
+    "test_map_Coverall": 0.6296,
+    "test_map_Face_Shield": 0.4194,
+    "test_map_Gloves": 0.3438,
+    "test_map_Goggles": 0.301,
+    "test_map_Mask": 0.4891,
+    "test_map_large": 0.6324,
+    "test_map_medium": 0.2898,
+    "test_map_small": 0.4357,
+    "test_mar_1": 0.3442,
+    "test_mar_10": 0.5781,
+    "test_mar_100": 0.6031,
+    "test_mar_100_Coverall": 0.7222,
+    "test_mar_100_Face_Shield": 0.6824,
+    "test_mar_100_Gloves": 0.5016,
+    "test_mar_100_Goggles": 0.5344,
+    "test_mar_100_Mask": 0.575,
+    "test_mar_large": 0.7618,
+    "test_mar_medium": 0.4489,
+    "test_mar_small": 0.5004,
+    "test_runtime": 3.552,
+    "test_samples_per_second": 8.164,
+    "test_steps_per_second": 1.126,
+    "total_flos": 2.0297599094333424e+19,
+    "train_loss": 1.1465214905648862,
+    "train_runtime": 16105.6449,
+    "train_samples_per_second": 5.278,
+    "train_steps_per_second": 0.658
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 99.53051643192488,
+    "eval_loss": 1.101932406425476,
+    "eval_map": 0.298,
+    "eval_map_50": 0.5545,
+    "eval_map_75": 0.2748,
+    "eval_map_Coverall": 0.5405,
+    "eval_map_Face_Shield": 0.2887,
+    "eval_map_Gloves": 0.2012,
+    "eval_map_Goggles": 0.1644,
+    "eval_map_Mask": 0.2954,
+    "eval_map_large": 0.4637,
+    "eval_map_medium": 0.2403,
+    "eval_map_small": 0.1608,
+    "eval_mar_1": 0.2807,
+    "eval_mar_10": 0.469,
+    "eval_mar_100": 0.5006,
+    "eval_mar_100_Coverall": 0.6698,
+    "eval_mar_100_Face_Shield": 0.5089,
+    "eval_mar_100_Gloves": 0.4455,
+    "eval_mar_100_Goggles": 0.4185,
+    "eval_mar_100_Mask": 0.4604,
+    "eval_mar_large": 0.6962,
+    "eval_mar_medium": 0.4532,
+    "eval_mar_small": 0.3003,
+    "eval_runtime": 15.523,
+    "eval_samples_per_second": 9.663,
+    "eval_steps_per_second": 1.224
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 99.53051643192488,
+    "test_loss": 0.9166296720504761,
+    "test_map": 0.4366,
+    "test_map_50": 0.7304,
+    "test_map_75": 0.4448,
+    "test_map_Coverall": 0.6296,
+    "test_map_Face_Shield": 0.4194,
+    "test_map_Gloves": 0.3438,
+    "test_map_Goggles": 0.301,
+    "test_map_Mask": 0.4891,
+    "test_map_large": 0.6324,
+    "test_map_medium": 0.2898,
+    "test_map_small": 0.4357,
+    "test_mar_1": 0.3442,
+    "test_mar_10": 0.5781,
+    "test_mar_100": 0.6031,
+    "test_mar_100_Coverall": 0.7222,
+    "test_mar_100_Face_Shield": 0.6824,
+    "test_mar_100_Gloves": 0.5016,
+    "test_mar_100_Goggles": 0.5344,
+    "test_mar_100_Mask": 0.575,
+    "test_mar_large": 0.7618,
+    "test_mar_medium": 0.4489,
+    "test_mar_small": 0.5004,
+    "test_runtime": 3.552,
+    "test_samples_per_second": 8.164,
+    "test_steps_per_second": 1.126
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 99.53051643192488,
+    "total_flos": 2.0297599094333424e+19,
+    "train_loss": 1.1465214905648862,
+    "train_runtime": 16105.6449,
+    "train_samples_per_second": 5.278,
+    "train_steps_per_second": 0.658
+}