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-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/01a30203)
 # facebook-detr-resnet-50-finetuned-10k-cppe5-manual-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.3496
-- Map: 0.2868
-- Map 50: 0.5691
-- Map 75: 0.2516
-- Map Small: 0.1306
-- Map Medium: 0.2195
-- Map Large: 0.5019
-- Mar 1: 0.2862
-- Mar 10: 0.445
-- Mar 100: 0.4563
-- Mar Small: 0.2314
-- Mar Medium: 0.3882
-- Mar Large: 0.6898
-- Map Coverall: 0.544
-- Mar 100 Coverall: 0.6635
-- Map Face Shield: 0.236
-- Mar 100 Face Shield: 0.462
-- Map Gloves: 0.2212
-- Mar 100 Gloves: 0.3888
-- Map Goggles: 0.1528
-- Mar 100 Goggles: 0.3738
-- Map Mask: 0.2799
-- Mar 100 Mask: 0.3933
+- Loss: 1.3392
+- Map: 0.291
+- Map 50: 0.568
+- Map 75: 0.2641
+- Map Small: 0.1359
+- Map Medium: 0.2289
+- Map Large: 0.4946
+- Mar 1: 0.2894
+- Mar 10: 0.4481
+- Mar 100: 0.4582
+- Mar Small: 0.2383
+- Mar Medium: 0.4004
+- Mar Large: 0.6871
+- Map Coverall: 0.5463
+- Mar 100 Coverall: 0.6676
+- Map Face Shield: 0.2454
+- Mar 100 Face Shield: 0.4405
+- Map Gloves: 0.216
+- Mar 100 Gloves: 0.3924
+- Map Goggles: 0.1691
+- Mar 100 Goggles: 0.3969
+- Map Mask: 0.278
+- Mar 100 Mask: 0.3938
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.3391669988632202,
+    "eval_map": 0.291,
+    "eval_map_50": 0.568,
+    "eval_map_75": 0.2641,
+    "eval_map_Coverall": 0.5463,
+    "eval_map_Face_Shield": 0.2454,
+    "eval_map_Gloves": 0.216,
+    "eval_map_Goggles": 0.1691,
+    "eval_map_Mask": 0.278,
+    "eval_map_large": 0.4946,
+    "eval_map_medium": 0.2289,
+    "eval_map_small": 0.1359,
+    "eval_mar_1": 0.2894,
+    "eval_mar_10": 0.4481,
+    "eval_mar_100": 0.4582,
+    "eval_mar_100_Coverall": 0.6676,
+    "eval_mar_100_Face_Shield": 0.4405,
+    "eval_mar_100_Gloves": 0.3924,
+    "eval_mar_100_Goggles": 0.3969,
+    "eval_mar_100_Mask": 0.3938,
+    "eval_mar_large": 0.6871,
+    "eval_mar_medium": 0.4004,
+    "eval_mar_small": 0.2383,
+    "eval_runtime": 6.9883,
+    "eval_samples_per_second": 21.464,
+    "eval_steps_per_second": 2.719,
+    "test_loss": 1.0454223155975342,
+    "test_map": 0.4602,
+    "test_map_50": 0.8147,
+    "test_map_75": 0.4515,
+    "test_map_Coverall": 0.6633,
+    "test_map_Face_Shield": 0.5231,
+    "test_map_Gloves": 0.3942,
+    "test_map_Goggles": 0.2844,
+    "test_map_Mask": 0.4361,
+    "test_map_large": 0.6894,
+    "test_map_medium": 0.3479,
+    "test_map_small": 0.3561,
+    "test_mar_1": 0.3531,
+    "test_mar_10": 0.5695,
+    "test_mar_100": 0.5755,
+    "test_mar_100_Coverall": 0.7533,
+    "test_mar_100_Face_Shield": 0.6765,
+    "test_mar_100_Gloves": 0.4721,
+    "test_mar_100_Goggles": 0.4563,
+    "test_mar_100_Mask": 0.5192,
+    "test_mar_large": 0.7381,
+    "test_mar_medium": 0.4441,
+    "test_mar_small": 0.4885,
+    "test_runtime": 1.9771,
+    "test_samples_per_second": 14.668,
+    "test_steps_per_second": 2.023,
+    "total_flos": 2.28455679528e+19,
+    "train_loss": 0.9875392821124781,
+    "train_runtime": 5219.9382,
+    "train_samples_per_second": 16.284,
+    "train_steps_per_second": 2.05
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.3391669988632202,
+    "eval_map": 0.291,
+    "eval_map_50": 0.568,
+    "eval_map_75": 0.2641,
+    "eval_map_Coverall": 0.5463,
+    "eval_map_Face_Shield": 0.2454,
+    "eval_map_Gloves": 0.216,
+    "eval_map_Goggles": 0.1691,
+    "eval_map_Mask": 0.278,
+    "eval_map_large": 0.4946,
+    "eval_map_medium": 0.2289,
+    "eval_map_small": 0.1359,
+    "eval_mar_1": 0.2894,
+    "eval_mar_10": 0.4481,
+    "eval_mar_100": 0.4582,
+    "eval_mar_100_Coverall": 0.6676,
+    "eval_mar_100_Face_Shield": 0.4405,
+    "eval_mar_100_Gloves": 0.3924,
+    "eval_mar_100_Goggles": 0.3969,
+    "eval_mar_100_Mask": 0.3938,
+    "eval_mar_large": 0.6871,
+    "eval_mar_medium": 0.4004,
+    "eval_mar_small": 0.2383,
+    "eval_runtime": 6.9883,
+    "eval_samples_per_second": 21.464,
+    "eval_steps_per_second": 2.719
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "test_loss": 1.0454223155975342,
+    "test_map": 0.4602,
+    "test_map_50": 0.8147,
+    "test_map_75": 0.4515,
+    "test_map_Coverall": 0.6633,
+    "test_map_Face_Shield": 0.5231,
+    "test_map_Gloves": 0.3942,
+    "test_map_Goggles": 0.2844,
+    "test_map_Mask": 0.4361,
+    "test_map_large": 0.6894,
+    "test_map_medium": 0.3479,
+    "test_map_small": 0.3561,
+    "test_mar_1": 0.3531,
+    "test_mar_10": 0.5695,
+    "test_mar_100": 0.5755,
+    "test_mar_100_Coverall": 0.7533,
+    "test_mar_100_Face_Shield": 0.6765,
+    "test_mar_100_Gloves": 0.4721,
+    "test_mar_100_Goggles": 0.4563,
+    "test_mar_100_Mask": 0.5192,
+    "test_mar_large": 0.7381,
+    "test_mar_medium": 0.4441,
+    "test_mar_small": 0.4885,
+    "test_runtime": 1.9771,
+    "test_samples_per_second": 14.668,
+    "test_steps_per_second": 2.023
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 100.0,
+    "total_flos": 2.28455679528e+19,
+    "train_loss": 0.9875392821124781,
+    "train_runtime": 5219.9382,
+    "train_samples_per_second": 16.284,
+    "train_steps_per_second": 2.05
+}