End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: facebook/detr-resnet-50
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: facebook-detr-resnet-50-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/x8wu1piw)
 # facebook-detr-resnet-50-finetuned-10k-cppe5-auto-pad
 
-This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on an unknown dataset.
+This model is a fine-tuned version of [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.3664
-- Map: 0.2799
-- Map 50: 0.5839
-- Map 75: 0.236
-- Map Small: 0.1569
+- Loss: 1.3763
+- Map: 0.2829
+- Map 50: 0.5916
+- Map 75: 0.2359
+- Map Small: 0.1623
 - Map Medium: 0.2077
-- Map Large: 0.4747
-- Mar 1: 0.2825
-- Mar 10: 0.4423
-- Mar 100: 0.4532
-- Mar Small: 0.2308
-- Mar Medium: 0.3887
-- Mar Large: 0.69
-- Map Coverall: 0.5471
-- Mar 100 Coverall: 0.6811
-- Map Face Shield: 0.2937
-- Mar 100 Face Shield: 0.4684
-- Map Gloves: 0.1975
-- Mar 100 Gloves: 0.3817
+- Map Large: 0.4793
+- Mar 1: 0.2809
+- Mar 10: 0.4454
+- Mar 100: 0.4571
+- Mar Small: 0.2381
+- Mar Medium: 0.3921
+- Mar Large: 0.6929
+- Map Coverall: 0.5515
+- Mar 100 Coverall: 0.6829
+- Map Face Shield: 0.2952
+- Mar 100 Face Shield: 0.4734
+- Map Gloves: 0.2003
+- Mar 100 Gloves: 0.3826
 - Map Goggles: 0.1173
-- Mar 100 Goggles: 0.3723
-- Map Mask: 0.2438
-- Mar 100 Mask: 0.3627
+- Mar 100 Goggles: 0.3815
+- Map Mask: 0.25
+- Mar 100 Mask: 0.3649
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.3763102293014526,
+    "eval_map": 0.2829,
+    "eval_map_50": 0.5916,
+    "eval_map_75": 0.2359,
+    "eval_map_Coverall": 0.5515,
+    "eval_map_Face_Shield": 0.2952,
+    "eval_map_Gloves": 0.2003,
+    "eval_map_Goggles": 0.1173,
+    "eval_map_Mask": 0.25,
+    "eval_map_large": 0.4793,
+    "eval_map_medium": 0.2077,
+    "eval_map_small": 0.1623,
+    "eval_mar_1": 0.2809,
+    "eval_mar_10": 0.4454,
+    "eval_mar_100": 0.4571,
+    "eval_mar_100_Coverall": 0.6829,
+    "eval_mar_100_Face_Shield": 0.4734,
+    "eval_mar_100_Gloves": 0.3826,
+    "eval_mar_100_Goggles": 0.3815,
+    "eval_mar_100_Mask": 0.3649,
+    "eval_mar_large": 0.6929,
+    "eval_mar_medium": 0.3921,
+    "eval_mar_small": 0.2381,
+    "eval_runtime": 5.9225,
+    "eval_samples_per_second": 25.327,
+    "eval_steps_per_second": 3.208,
+    "test_loss": 1.2338895797729492,
+    "test_map": 0.3953,
+    "test_map_50": 0.7479,
+    "test_map_75": 0.3573,
+    "test_map_Coverall": 0.6538,
+    "test_map_Face_Shield": 0.4816,
+    "test_map_Gloves": 0.2747,
+    "test_map_Goggles": 0.2368,
+    "test_map_Mask": 0.3298,
+    "test_map_large": 0.6567,
+    "test_map_medium": 0.2724,
+    "test_map_small": 0.3566,
+    "test_mar_1": 0.3197,
+    "test_mar_10": 0.5407,
+    "test_mar_100": 0.5559,
+    "test_mar_100_Coverall": 0.7556,
+    "test_mar_100_Face_Shield": 0.7,
+    "test_mar_100_Gloves": 0.4459,
+    "test_mar_100_Goggles": 0.4375,
+    "test_mar_100_Mask": 0.4404,
+    "test_mar_large": 0.793,
+    "test_mar_medium": 0.388,
+    "test_mar_small": 0.4374,
+    "test_runtime": 1.9935,
+    "test_samples_per_second": 14.547,
+    "test_steps_per_second": 2.007,
+    "total_flos": 2.2505492746344178e+19,
+    "train_loss": 1.1716438714143271,
+    "train_runtime": 4951.8709,
+    "train_samples_per_second": 17.165,
+    "train_steps_per_second": 2.161
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.3763102293014526,
+    "eval_map": 0.2829,
+    "eval_map_50": 0.5916,
+    "eval_map_75": 0.2359,
+    "eval_map_Coverall": 0.5515,
+    "eval_map_Face_Shield": 0.2952,
+    "eval_map_Gloves": 0.2003,
+    "eval_map_Goggles": 0.1173,
+    "eval_map_Mask": 0.25,
+    "eval_map_large": 0.4793,
+    "eval_map_medium": 0.2077,
+    "eval_map_small": 0.1623,
+    "eval_mar_1": 0.2809,
+    "eval_mar_10": 0.4454,
+    "eval_mar_100": 0.4571,
+    "eval_mar_100_Coverall": 0.6829,
+    "eval_mar_100_Face_Shield": 0.4734,
+    "eval_mar_100_Gloves": 0.3826,
+    "eval_mar_100_Goggles": 0.3815,
+    "eval_mar_100_Mask": 0.3649,
+    "eval_mar_large": 0.6929,
+    "eval_mar_medium": 0.3921,
+    "eval_mar_small": 0.2381,
+    "eval_runtime": 5.9225,
+    "eval_samples_per_second": 25.327,
+    "eval_steps_per_second": 3.208
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "test_loss": 1.2338895797729492,
+    "test_map": 0.3953,
+    "test_map_50": 0.7479,
+    "test_map_75": 0.3573,
+    "test_map_Coverall": 0.6538,
+    "test_map_Face_Shield": 0.4816,
+    "test_map_Gloves": 0.2747,
+    "test_map_Goggles": 0.2368,
+    "test_map_Mask": 0.3298,
+    "test_map_large": 0.6567,
+    "test_map_medium": 0.2724,
+    "test_map_small": 0.3566,
+    "test_mar_1": 0.3197,
+    "test_mar_10": 0.5407,
+    "test_mar_100": 0.5559,
+    "test_mar_100_Coverall": 0.7556,
+    "test_mar_100_Face_Shield": 0.7,
+    "test_mar_100_Gloves": 0.4459,
+    "test_mar_100_Goggles": 0.4375,
+    "test_mar_100_Mask": 0.4404,
+    "test_mar_large": 0.793,
+    "test_mar_medium": 0.388,
+    "test_mar_small": 0.4374,
+    "test_runtime": 1.9935,
+    "test_samples_per_second": 14.547,
+    "test_steps_per_second": 2.007
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 100.0,
+    "total_flos": 2.2505492746344178e+19,
+    "train_loss": 1.1716438714143271,
+    "train_runtime": 4951.8709,
+    "train_samples_per_second": 17.165,
+    "train_steps_per_second": 2.161
+}