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-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/8dms9fij)
 # microsoft-conditional-detr-resnet-50-finetuned-10k-cppe5-auto-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.3455
-- Map: 0.2991
-- Map 50: 0.6019
-- Map 75: 0.2518
-- Map Small: 0.1694
-- Map Medium: 0.2405
-- Map Large: 0.4915
-- Mar 1: 0.2856
-- Mar 10: 0.4211
-- Mar 100: 0.4384
-- Mar Small: 0.2598
-- Mar Medium: 0.389
-- Mar Large: 0.64
-- Map Coverall: 0.5526
-- Mar 100 Coverall: 0.6491
-- Map Face Shield: 0.3332
-- Mar 100 Face Shield: 0.4633
-- Map Gloves: 0.1783
-- Mar 100 Gloves: 0.3357
-- Map Goggles: 0.1552
-- Mar 100 Goggles: 0.3415
-- Map Mask: 0.2765
-- Mar 100 Mask: 0.4022
+- Loss: 1.3408
+- Map: 0.3032
+- Map 50: 0.5973
+- Map 75: 0.2676
+- Map Small: 0.1648
+- Map Medium: 0.2427
+- Map Large: 0.4981
+- Mar 1: 0.2761
+- Mar 10: 0.426
+- Mar 100: 0.4447
+- Mar Small: 0.2601
+- Mar Medium: 0.3835
+- Mar Large: 0.663
+- Map Coverall: 0.553
+- Mar 100 Coverall: 0.655
+- Map Face Shield: 0.3411
+- Mar 100 Face Shield: 0.4722
+- Map Gloves: 0.1825
+- Mar 100 Gloves: 0.3371
+- Map Goggles: 0.1574
+- Mar 100 Goggles: 0.3538
+- Map Mask: 0.282
+- Mar 100 Mask: 0.4053
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.340824842453003,
+    "eval_map": 0.3032,
+    "eval_map_50": 0.5973,
+    "eval_map_75": 0.2676,
+    "eval_map_Coverall": 0.553,
+    "eval_map_Face_Shield": 0.3411,
+    "eval_map_Gloves": 0.1825,
+    "eval_map_Goggles": 0.1574,
+    "eval_map_Mask": 0.282,
+    "eval_map_large": 0.4981,
+    "eval_map_medium": 0.2427,
+    "eval_map_small": 0.1648,
+    "eval_mar_1": 0.2761,
+    "eval_mar_10": 0.426,
+    "eval_mar_100": 0.4447,
+    "eval_mar_100_Coverall": 0.655,
+    "eval_mar_100_Face_Shield": 0.4722,
+    "eval_mar_100_Gloves": 0.3371,
+    "eval_mar_100_Goggles": 0.3538,
+    "eval_mar_100_Mask": 0.4053,
+    "eval_mar_large": 0.663,
+    "eval_mar_medium": 0.3835,
+    "eval_mar_small": 0.2601,
+    "eval_runtime": 6.2922,
+    "eval_samples_per_second": 23.839,
+    "eval_steps_per_second": 3.02,
+    "test_loss": 1.2219938039779663,
+    "test_map": 0.4162,
+    "test_map_50": 0.7489,
+    "test_map_75": 0.3848,
+    "test_map_Coverall": 0.6371,
+    "test_map_Face_Shield": 0.5697,
+    "test_map_Gloves": 0.2862,
+    "test_map_Goggles": 0.2241,
+    "test_map_Mask": 0.3638,
+    "test_map_large": 0.6303,
+    "test_map_medium": 0.2323,
+    "test_map_small": 0.4026,
+    "test_mar_1": 0.3173,
+    "test_mar_10": 0.5208,
+    "test_mar_100": 0.5371,
+    "test_mar_100_Coverall": 0.7156,
+    "test_mar_100_Face_Shield": 0.6941,
+    "test_mar_100_Gloves": 0.3967,
+    "test_mar_100_Goggles": 0.4062,
+    "test_mar_100_Mask": 0.4731,
+    "test_mar_large": 0.7364,
+    "test_mar_medium": 0.3612,
+    "test_mar_small": 0.47,
+    "test_runtime": 1.809,
+    "test_samples_per_second": 16.031,
+    "test_steps_per_second": 2.211,
+    "total_flos": 2.3505198523920773e+19,
+    "train_loss": 1.2746832000875028,
+    "train_runtime": 5432.0149,
+    "train_samples_per_second": 15.648,
+    "train_steps_per_second": 1.97
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.340824842453003,
+    "eval_map": 0.3032,
+    "eval_map_50": 0.5973,
+    "eval_map_75": 0.2676,
+    "eval_map_Coverall": 0.553,
+    "eval_map_Face_Shield": 0.3411,
+    "eval_map_Gloves": 0.1825,
+    "eval_map_Goggles": 0.1574,
+    "eval_map_Mask": 0.282,
+    "eval_map_large": 0.4981,
+    "eval_map_medium": 0.2427,
+    "eval_map_small": 0.1648,
+    "eval_mar_1": 0.2761,
+    "eval_mar_10": 0.426,
+    "eval_mar_100": 0.4447,
+    "eval_mar_100_Coverall": 0.655,
+    "eval_mar_100_Face_Shield": 0.4722,
+    "eval_mar_100_Gloves": 0.3371,
+    "eval_mar_100_Goggles": 0.3538,
+    "eval_mar_100_Mask": 0.4053,
+    "eval_mar_large": 0.663,
+    "eval_mar_medium": 0.3835,
+    "eval_mar_small": 0.2601,
+    "eval_runtime": 6.2922,
+    "eval_samples_per_second": 23.839,
+    "eval_steps_per_second": 3.02
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "test_loss": 1.2219938039779663,
+    "test_map": 0.4162,
+    "test_map_50": 0.7489,
+    "test_map_75": 0.3848,
+    "test_map_Coverall": 0.6371,
+    "test_map_Face_Shield": 0.5697,
+    "test_map_Gloves": 0.2862,
+    "test_map_Goggles": 0.2241,
+    "test_map_Mask": 0.3638,
+    "test_map_large": 0.6303,
+    "test_map_medium": 0.2323,
+    "test_map_small": 0.4026,
+    "test_mar_1": 0.3173,
+    "test_mar_10": 0.5208,
+    "test_mar_100": 0.5371,
+    "test_mar_100_Coverall": 0.7156,
+    "test_mar_100_Face_Shield": 0.6941,
+    "test_mar_100_Gloves": 0.3967,
+    "test_mar_100_Goggles": 0.4062,
+    "test_mar_100_Mask": 0.4731,
+    "test_mar_large": 0.7364,
+    "test_mar_medium": 0.3612,
+    "test_mar_small": 0.47,
+    "test_runtime": 1.809,
+    "test_samples_per_second": 16.031,
+    "test_steps_per_second": 2.211
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 100.0,
+    "total_flos": 2.3505198523920773e+19,
+    "train_loss": 1.2746832000875028,
+    "train_runtime": 5432.0149,
+    "train_samples_per_second": 15.648,
+    "train_steps_per_second": 1.97
+}