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,7 +16,7 @@ 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/sodr0hal)
 # 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

all_results.json

@@ -26,35 +26,35 @@
     "eval_runtime": 15.6632,
     "eval_samples_per_second": 9.577,
     "eval_steps_per_second": 1.213,
-    "test_loss": 1.2813646793365479,
-    "test_map": 0.3392,
-    "test_map_50": 0.8178,
-    "test_map_75": 0.1342,
-    "test_map_Coverall": 0.4236,
-    "test_map_Face_Shield": 0.366,
-    "test_map_Gloves": 0.3146,
-    "test_map_Goggles": 0.2668,
-    "test_map_Mask": 0.325,
-    "test_map_large": 0.4557,
-    "test_map_medium": 0.3312,
-    "test_map_small": 0.3208,
-    "test_mar_1": 0.2457,
-    "test_mar_10": 0.4543,
-    "test_mar_100": 0.4723,
-    "test_mar_100_Coverall": 0.5733,
-    "test_mar_100_Face_Shield": 0.5176,
-    "test_mar_100_Gloves": 0.4148,
-    "test_mar_100_Goggles": 0.425,
-    "test_mar_100_Mask": 0.4308,
-    "test_mar_large": 0.5709,
-    "test_mar_medium": 0.4222,
-    "test_mar_small": 0.3702,
-    "test_runtime": 5.3874,
-    "test_samples_per_second": 5.383,
-    "test_steps_per_second": 5.383,
+    "test_loss": 2.403155565261841,
+    "test_map": 0.0814,
+    "test_map_50": 0.2065,
+    "test_map_75": 0.0234,
+    "test_map_Coverall": 0.0796,
+    "test_map_Face_Shield": 0.2432,
+    "test_map_Gloves": 0.0278,
+    "test_map_Goggles": 0.0116,
+    "test_map_Mask": 0.0447,
+    "test_map_large": 0.1559,
+    "test_map_medium": 0.0718,
+    "test_map_small": 0.0144,
+    "test_mar_1": 0.1018,
+    "test_mar_10": 0.1983,
+    "test_mar_100": 0.2149,
+    "test_mar_100_Coverall": 0.2822,
+    "test_mar_100_Face_Shield": 0.3941,
+    "test_mar_100_Gloves": 0.1246,
+    "test_mar_100_Goggles": 0.1219,
+    "test_mar_100_Mask": 0.1519,
+    "test_mar_large": 0.3267,
+    "test_mar_medium": 0.1528,
+    "test_mar_small": 0.0427,
+    "test_runtime": 4.5658,
+    "test_samples_per_second": 6.352,
+    "test_steps_per_second": 0.876,
     "total_flos": 2.1870706694064443e+19,
     "train_loss": 0.0,
-    "train_runtime": 2.9311,
-    "train_samples_per_second": 28998.867,
-    "train_steps_per_second": 3616.329
+    "train_runtime": 3.0991,
+    "train_samples_per_second": 27427.639,
+    "train_steps_per_second": 3420.388
 }

test_results.json

@@ -1,29 +1,29 @@
 {
     "epoch": 99.53051643192488,
-    "test_loss": 1.2813646793365479,
-    "test_map": 0.3392,
-    "test_map_50": 0.8178,
-    "test_map_75": 0.1342,
-    "test_map_Coverall": 0.4236,
-    "test_map_Face_Shield": 0.366,
-    "test_map_Gloves": 0.3146,
-    "test_map_Goggles": 0.2668,
-    "test_map_Mask": 0.325,
-    "test_map_large": 0.4557,
-    "test_map_medium": 0.3312,
-    "test_map_small": 0.3208,
-    "test_mar_1": 0.2457,
-    "test_mar_10": 0.4543,
-    "test_mar_100": 0.4723,
-    "test_mar_100_Coverall": 0.5733,
-    "test_mar_100_Face_Shield": 0.5176,
-    "test_mar_100_Gloves": 0.4148,
-    "test_mar_100_Goggles": 0.425,
-    "test_mar_100_Mask": 0.4308,
-    "test_mar_large": 0.5709,
-    "test_mar_medium": 0.4222,
-    "test_mar_small": 0.3702,
-    "test_runtime": 5.3874,
-    "test_samples_per_second": 5.383,
-    "test_steps_per_second": 5.383
+    "test_loss": 2.403155565261841,
+    "test_map": 0.0814,
+    "test_map_50": 0.2065,
+    "test_map_75": 0.0234,
+    "test_map_Coverall": 0.0796,
+    "test_map_Face_Shield": 0.2432,
+    "test_map_Gloves": 0.0278,
+    "test_map_Goggles": 0.0116,
+    "test_map_Mask": 0.0447,
+    "test_map_large": 0.1559,
+    "test_map_medium": 0.0718,
+    "test_map_small": 0.0144,
+    "test_mar_1": 0.1018,
+    "test_mar_10": 0.1983,
+    "test_mar_100": 0.2149,
+    "test_mar_100_Coverall": 0.2822,
+    "test_mar_100_Face_Shield": 0.3941,
+    "test_mar_100_Gloves": 0.1246,
+    "test_mar_100_Goggles": 0.1219,
+    "test_mar_100_Mask": 0.1519,
+    "test_mar_large": 0.3267,
+    "test_mar_medium": 0.1528,
+    "test_mar_small": 0.0427,
+    "test_runtime": 4.5658,
+    "test_samples_per_second": 6.352,
+    "test_steps_per_second": 0.876
 }

train_results.json

@@ -2,7 +2,7 @@
     "epoch": 99.53051643192488,
     "total_flos": 2.1870706694064443e+19,
     "train_loss": 0.0,
-    "train_runtime": 2.9311,
-    "train_samples_per_second": 28998.867,
-    "train_steps_per_second": 3616.329
+    "train_runtime": 3.0991,
+    "train_samples_per_second": 27427.639,
+    "train_steps_per_second": 3420.388
 }

trainer_state.json

@@ -3713,9 +3713,9 @@
       "step": 10600,
       "total_flos": 2.1870706694064443e+19,
       "train_loss": 0.0,
-      "train_runtime": 2.9311,
-      "train_samples_per_second": 28998.867,
-      "train_steps_per_second": 3616.329
+      "train_runtime": 3.0991,
+      "train_samples_per_second": 27427.639,
+      "train_steps_per_second": 3420.388
     }
   ],
   "logging_steps": 500,