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-repro
@@ -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/y96s266q)
 # facebook-detr-resnet-50-finetuned-10k-cppe5-manual-pad-repro
 
-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.3550
-- Map: 0.2936
-- Map 50: 0.5835
-- Map 75: 0.2461
-- Map Small: 0.1635
-- Map Medium: 0.2284
-- Map Large: 0.4956
-- Mar 1: 0.2943
-- Mar 10: 0.4587
-- Mar 100: 0.4737
-- Mar Small: 0.2424
-- Mar Medium: 0.4204
-- Mar Large: 0.6907
-- Map Coverall: 0.5418
-- Mar 100 Coverall: 0.6788
-- Map Face Shield: 0.2801
-- Mar 100 Face Shield: 0.481
-- Map Gloves: 0.2075
-- Mar 100 Gloves: 0.3955
-- Map Goggles: 0.1534
-- Mar 100 Goggles: 0.4108
-- Map Mask: 0.2852
-- Mar 100 Mask: 0.4022
+- Loss: 1.3283
+- Map: 0.2947
+- Map 50: 0.5809
+- Map 75: 0.2681
+- Map Small: 0.1583
+- Map Medium: 0.23
+- Map Large: 0.4971
+- Mar 1: 0.2972
+- Mar 10: 0.4633
+- Mar 100: 0.4771
+- Mar Small: 0.2237
+- Mar Medium: 0.4317
+- Mar Large: 0.7008
+- Map Coverall: 0.5445
+- Mar 100 Coverall: 0.6829
+- Map Face Shield: 0.2753
+- Mar 100 Face Shield: 0.4937
+- Map Gloves: 0.2028
+- Mar 100 Gloves: 0.4098
+- Map Goggles: 0.151
+- Mar 100 Goggles: 0.3938
+- Map Mask: 0.3002
+- Mar 100 Mask: 0.4053
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.328338861465454,
+    "eval_map": 0.2947,
+    "eval_map_50": 0.5809,
+    "eval_map_75": 0.2681,
+    "eval_map_Coverall": 0.5445,
+    "eval_map_Face_Shield": 0.2753,
+    "eval_map_Gloves": 0.2028,
+    "eval_map_Goggles": 0.151,
+    "eval_map_Mask": 0.3002,
+    "eval_map_large": 0.4971,
+    "eval_map_medium": 0.23,
+    "eval_map_small": 0.1583,
+    "eval_mar_1": 0.2972,
+    "eval_mar_10": 0.4633,
+    "eval_mar_100": 0.4771,
+    "eval_mar_100_Coverall": 0.6829,
+    "eval_mar_100_Face_Shield": 0.4937,
+    "eval_mar_100_Gloves": 0.4098,
+    "eval_mar_100_Goggles": 0.3938,
+    "eval_mar_100_Mask": 0.4053,
+    "eval_mar_large": 0.7008,
+    "eval_mar_medium": 0.4317,
+    "eval_mar_small": 0.2237,
+    "eval_runtime": 6.3011,
+    "eval_samples_per_second": 23.805,
+    "eval_steps_per_second": 3.015,
+    "test_loss": 1.0668703317642212,
+    "test_map": 0.4516,
+    "test_map_50": 0.7874,
+    "test_map_75": 0.4614,
+    "test_map_Coverall": 0.6417,
+    "test_map_Face_Shield": 0.536,
+    "test_map_Gloves": 0.3737,
+    "test_map_Goggles": 0.293,
+    "test_map_Mask": 0.4137,
+    "test_map_large": 0.6744,
+    "test_map_medium": 0.3415,
+    "test_map_small": 0.4057,
+    "test_mar_1": 0.3473,
+    "test_mar_10": 0.5856,
+    "test_mar_100": 0.6007,
+    "test_mar_100_Coverall": 0.7511,
+    "test_mar_100_Face_Shield": 0.7176,
+    "test_mar_100_Gloves": 0.4869,
+    "test_mar_100_Goggles": 0.525,
+    "test_mar_100_Mask": 0.5231,
+    "test_mar_large": 0.8051,
+    "test_mar_medium": 0.492,
+    "test_mar_small": 0.4962,
+    "test_runtime": 1.812,
+    "test_samples_per_second": 16.005,
+    "test_steps_per_second": 2.208,
+    "total_flos": 2.28455679528e+19,
+    "train_loss": 0.9858380618942119,
+    "train_runtime": 5956.9501,
+    "train_samples_per_second": 14.269,
+    "train_steps_per_second": 1.796
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.328338861465454,
+    "eval_map": 0.2947,
+    "eval_map_50": 0.5809,
+    "eval_map_75": 0.2681,
+    "eval_map_Coverall": 0.5445,
+    "eval_map_Face_Shield": 0.2753,
+    "eval_map_Gloves": 0.2028,
+    "eval_map_Goggles": 0.151,
+    "eval_map_Mask": 0.3002,
+    "eval_map_large": 0.4971,
+    "eval_map_medium": 0.23,
+    "eval_map_small": 0.1583,
+    "eval_mar_1": 0.2972,
+    "eval_mar_10": 0.4633,
+    "eval_mar_100": 0.4771,
+    "eval_mar_100_Coverall": 0.6829,
+    "eval_mar_100_Face_Shield": 0.4937,
+    "eval_mar_100_Gloves": 0.4098,
+    "eval_mar_100_Goggles": 0.3938,
+    "eval_mar_100_Mask": 0.4053,
+    "eval_mar_large": 0.7008,
+    "eval_mar_medium": 0.4317,
+    "eval_mar_small": 0.2237,
+    "eval_runtime": 6.3011,
+    "eval_samples_per_second": 23.805,
+    "eval_steps_per_second": 3.015
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "test_loss": 1.0668703317642212,
+    "test_map": 0.4516,
+    "test_map_50": 0.7874,
+    "test_map_75": 0.4614,
+    "test_map_Coverall": 0.6417,
+    "test_map_Face_Shield": 0.536,
+    "test_map_Gloves": 0.3737,
+    "test_map_Goggles": 0.293,
+    "test_map_Mask": 0.4137,
+    "test_map_large": 0.6744,
+    "test_map_medium": 0.3415,
+    "test_map_small": 0.4057,
+    "test_mar_1": 0.3473,
+    "test_mar_10": 0.5856,
+    "test_mar_100": 0.6007,
+    "test_mar_100_Coverall": 0.7511,
+    "test_mar_100_Face_Shield": 0.7176,
+    "test_mar_100_Gloves": 0.4869,
+    "test_mar_100_Goggles": 0.525,
+    "test_mar_100_Mask": 0.5231,
+    "test_mar_large": 0.8051,
+    "test_mar_medium": 0.492,
+    "test_mar_small": 0.4962,
+    "test_runtime": 1.812,
+    "test_samples_per_second": 16.005,
+    "test_steps_per_second": 2.208
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 100.0,
+    "total_flos": 2.28455679528e+19,
+    "train_loss": 0.9858380618942119,
+    "train_runtime": 5956.9501,
+    "train_samples_per_second": 14.269,
+    "train_steps_per_second": 1.796
+}