End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: microsoft/conditional-detr-resnet-50
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: microsoft-conditional-detr-resnet-50-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/ustfoyri)
 # microsoft-conditional-detr-resnet-50-finetuned-10k-cppe5-manual-pad
 
-This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on an unknown dataset.
+This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.3721
-- Map: 0.3104
-- Map 50: 0.6106
-- Map 75: 0.2705
-- Map Small: 0.1755
-- Map Medium: 0.2484
-- Map Large: 0.5049
-- Mar 1: 0.2951
-- Mar 10: 0.4381
-- Mar 100: 0.4488
-- Mar Small: 0.2478
-- Mar Medium: 0.3997
-- Mar Large: 0.6319
-- Map Coverall: 0.5609
-- Mar 100 Coverall: 0.6473
-- Map Face Shield: 0.3402
-- Mar 100 Face Shield: 0.4785
-- Map Gloves: 0.1979
-- Mar 100 Gloves: 0.3424
-- Map Goggles: 0.1636
-- Mar 100 Goggles: 0.3569
-- Map Mask: 0.2893
-- Mar 100 Mask: 0.4187
+- Loss: 1.2867
+- Map: 0.3142
+- Map 50: 0.6067
+- Map 75: 0.2769
+- Map Small: 0.1689
+- Map Medium: 0.2662
+- Map Large: 0.5119
+- Mar 1: 0.3035
+- Mar 10: 0.4412
+- Mar 100: 0.4638
+- Mar Small: 0.2652
+- Mar Medium: 0.4191
+- Mar Large: 0.6548
+- Map Coverall: 0.5532
+- Mar 100 Coverall: 0.6559
+- Map Face Shield: 0.3268
+- Mar 100 Face Shield: 0.4886
+- Map Gloves: 0.1937
+- Mar 100 Gloves: 0.3603
+- Map Goggles: 0.1922
+- Mar 100 Goggles: 0.3892
+- Map Mask: 0.3053
+- Mar 100 Mask: 0.4249
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.2867472171783447,
+    "eval_map": 0.3142,
+    "eval_map_50": 0.6067,
+    "eval_map_75": 0.2769,
+    "eval_map_Coverall": 0.5532,
+    "eval_map_Face_Shield": 0.3268,
+    "eval_map_Gloves": 0.1937,
+    "eval_map_Goggles": 0.1922,
+    "eval_map_Mask": 0.3053,
+    "eval_map_large": 0.5119,
+    "eval_map_medium": 0.2662,
+    "eval_map_small": 0.1689,
+    "eval_mar_1": 0.3035,
+    "eval_mar_10": 0.4412,
+    "eval_mar_100": 0.4638,
+    "eval_mar_100_Coverall": 0.6559,
+    "eval_mar_100_Face_Shield": 0.4886,
+    "eval_mar_100_Gloves": 0.3603,
+    "eval_mar_100_Goggles": 0.3892,
+    "eval_mar_100_Mask": 0.4249,
+    "eval_mar_large": 0.6548,
+    "eval_mar_medium": 0.4191,
+    "eval_mar_small": 0.2652,
+    "eval_runtime": 6.5804,
+    "eval_samples_per_second": 22.795,
+    "eval_steps_per_second": 2.887,
+    "test_loss": 1.0588135719299316,
+    "test_map": 0.4294,
+    "test_map_50": 0.8118,
+    "test_map_75": 0.3741,
+    "test_map_Coverall": 0.6132,
+    "test_map_Face_Shield": 0.5433,
+    "test_map_Gloves": 0.3281,
+    "test_map_Goggles": 0.2515,
+    "test_map_Mask": 0.4107,
+    "test_map_large": 0.6842,
+    "test_map_medium": 0.286,
+    "test_map_small": 0.4334,
+    "test_mar_1": 0.3424,
+    "test_mar_10": 0.5407,
+    "test_mar_100": 0.5501,
+    "test_mar_100_Coverall": 0.7022,
+    "test_mar_100_Face_Shield": 0.6647,
+    "test_mar_100_Gloves": 0.4393,
+    "test_mar_100_Goggles": 0.4344,
+    "test_mar_100_Mask": 0.5096,
+    "test_mar_large": 0.7515,
+    "test_mar_medium": 0.4006,
+    "test_mar_small": 0.4735,
+    "test_runtime": 2.1332,
+    "test_samples_per_second": 13.594,
+    "test_steps_per_second": 1.875,
+    "total_flos": 2.386038004919996e+19,
+    "train_loss": 0.934452736756512,
+    "train_runtime": 5543.1729,
+    "train_samples_per_second": 15.334,
+    "train_steps_per_second": 1.93
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.2867472171783447,
+    "eval_map": 0.3142,
+    "eval_map_50": 0.6067,
+    "eval_map_75": 0.2769,
+    "eval_map_Coverall": 0.5532,
+    "eval_map_Face_Shield": 0.3268,
+    "eval_map_Gloves": 0.1937,
+    "eval_map_Goggles": 0.1922,
+    "eval_map_Mask": 0.3053,
+    "eval_map_large": 0.5119,
+    "eval_map_medium": 0.2662,
+    "eval_map_small": 0.1689,
+    "eval_mar_1": 0.3035,
+    "eval_mar_10": 0.4412,
+    "eval_mar_100": 0.4638,
+    "eval_mar_100_Coverall": 0.6559,
+    "eval_mar_100_Face_Shield": 0.4886,
+    "eval_mar_100_Gloves": 0.3603,
+    "eval_mar_100_Goggles": 0.3892,
+    "eval_mar_100_Mask": 0.4249,
+    "eval_mar_large": 0.6548,
+    "eval_mar_medium": 0.4191,
+    "eval_mar_small": 0.2652,
+    "eval_runtime": 6.5804,
+    "eval_samples_per_second": 22.795,
+    "eval_steps_per_second": 2.887
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "test_loss": 1.0588135719299316,
+    "test_map": 0.4294,
+    "test_map_50": 0.8118,
+    "test_map_75": 0.3741,
+    "test_map_Coverall": 0.6132,
+    "test_map_Face_Shield": 0.5433,
+    "test_map_Gloves": 0.3281,
+    "test_map_Goggles": 0.2515,
+    "test_map_Mask": 0.4107,
+    "test_map_large": 0.6842,
+    "test_map_medium": 0.286,
+    "test_map_small": 0.4334,
+    "test_mar_1": 0.3424,
+    "test_mar_10": 0.5407,
+    "test_mar_100": 0.5501,
+    "test_mar_100_Coverall": 0.7022,
+    "test_mar_100_Face_Shield": 0.6647,
+    "test_mar_100_Gloves": 0.4393,
+    "test_mar_100_Goggles": 0.4344,
+    "test_mar_100_Mask": 0.5096,
+    "test_mar_large": 0.7515,
+    "test_mar_medium": 0.4006,
+    "test_mar_small": 0.4735,
+    "test_runtime": 2.1332,
+    "test_samples_per_second": 13.594,
+    "test_steps_per_second": 1.875
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 100.0,
+    "total_flos": 2.386038004919996e+19,
+    "train_loss": 0.934452736756512,
+    "train_runtime": 5543.1729,
+    "train_samples_per_second": 15.334,
+    "train_steps_per_second": 1.93
+}