End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: hustvl/yolos-small
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: hustvl-yolos-small-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/wew6j0zh)
 # hustvl-yolos-small-finetuned-10k-cppe5-manual-pad
 
-This model is a fine-tuned version of [hustvl/yolos-small](https://huggingface.co/hustvl/yolos-small) on an unknown dataset.
+This model is a fine-tuned version of [hustvl/yolos-small](https://huggingface.co/hustvl/yolos-small) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.8651
-- Map: 0.2893
-- Map 50: 0.5857
-- Map 75: 0.2362
-- Map Small: 0.162
-- Map Medium: 0.2199
-- Map Large: 0.4909
-- Mar 1: 0.2834
-- Mar 10: 0.4333
-- Mar 100: 0.4526
-- Mar Small: 0.2365
-- Mar Medium: 0.3956
-- Mar Large: 0.6923
-- Map Coverall: 0.549
+- Loss: 1.7972
+- Map: 0.292
+- Map 50: 0.5823
+- Map 75: 0.2547
+- Map Small: 0.1715
+- Map Medium: 0.2168
+- Map Large: 0.5148
+- Mar 1: 0.2804
+- Mar 10: 0.424
+- Mar 100: 0.4447
+- Mar Small: 0.224
+- Mar Medium: 0.3795
+- Mar Large: 0.6949
+- Map Coverall: 0.5577
 - Mar 100 Coverall: 0.6631
-- Map Face Shield: 0.2574
-- Mar 100 Face Shield: 0.438
-- Map Gloves: 0.1941
-- Mar 100 Gloves: 0.3969
-- Map Goggles: 0.1911
-- Mar 100 Goggles: 0.3769
-- Map Mask: 0.2551
-- Mar 100 Mask: 0.388
+- Map Face Shield: 0.2464
+- Mar 100 Face Shield: 0.4152
+- Map Gloves: 0.187
+- Mar 100 Gloves: 0.3964
+- Map Goggles: 0.2104
+- Mar 100 Goggles: 0.3615
+- Map Mask: 0.2586
+- Mar 100 Mask: 0.3871
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.7971552610397339,
+    "eval_map": 0.292,
+    "eval_map_50": 0.5823,
+    "eval_map_75": 0.2547,
+    "eval_map_Coverall": 0.5577,
+    "eval_map_Face_Shield": 0.2464,
+    "eval_map_Gloves": 0.187,
+    "eval_map_Goggles": 0.2104,
+    "eval_map_Mask": 0.2586,
+    "eval_map_large": 0.5148,
+    "eval_map_medium": 0.2168,
+    "eval_map_small": 0.1715,
+    "eval_mar_1": 0.2804,
+    "eval_mar_10": 0.424,
+    "eval_mar_100": 0.4447,
+    "eval_mar_100_Coverall": 0.6631,
+    "eval_mar_100_Face_Shield": 0.4152,
+    "eval_mar_100_Gloves": 0.3964,
+    "eval_mar_100_Goggles": 0.3615,
+    "eval_mar_100_Mask": 0.3871,
+    "eval_mar_large": 0.6949,
+    "eval_mar_medium": 0.3795,
+    "eval_mar_small": 0.224,
+    "eval_runtime": 7.7617,
+    "eval_samples_per_second": 19.326,
+    "eval_steps_per_second": 2.448,
+    "test_loss": 1.587841272354126,
+    "test_map": 0.3942,
+    "test_map_50": 0.7048,
+    "test_map_75": 0.3809,
+    "test_map_Coverall": 0.5521,
+    "test_map_Face_Shield": 0.575,
+    "test_map_Gloves": 0.3106,
+    "test_map_Goggles": 0.1884,
+    "test_map_Mask": 0.3449,
+    "test_map_large": 0.6081,
+    "test_map_medium": 0.2404,
+    "test_map_small": 0.422,
+    "test_mar_1": 0.3136,
+    "test_mar_10": 0.5138,
+    "test_mar_100": 0.5274,
+    "test_mar_100_Coverall": 0.6711,
+    "test_mar_100_Face_Shield": 0.7118,
+    "test_mar_100_Gloves": 0.4459,
+    "test_mar_100_Goggles": 0.3313,
+    "test_mar_100_Mask": 0.4769,
+    "test_mar_large": 0.7511,
+    "test_mar_medium": 0.4155,
+    "test_mar_small": 0.4541,
+    "test_runtime": 2.2158,
+    "test_samples_per_second": 13.088,
+    "test_steps_per_second": 1.805,
+    "total_flos": 1.643572308404736e+19,
+    "train_loss": 0.5523391162123635,
+    "train_runtime": 7636.311,
+    "train_samples_per_second": 11.131,
+    "train_steps_per_second": 1.401
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.7971552610397339,
+    "eval_map": 0.292,
+    "eval_map_50": 0.5823,
+    "eval_map_75": 0.2547,
+    "eval_map_Coverall": 0.5577,
+    "eval_map_Face_Shield": 0.2464,
+    "eval_map_Gloves": 0.187,
+    "eval_map_Goggles": 0.2104,
+    "eval_map_Mask": 0.2586,
+    "eval_map_large": 0.5148,
+    "eval_map_medium": 0.2168,
+    "eval_map_small": 0.1715,
+    "eval_mar_1": 0.2804,
+    "eval_mar_10": 0.424,
+    "eval_mar_100": 0.4447,
+    "eval_mar_100_Coverall": 0.6631,
+    "eval_mar_100_Face_Shield": 0.4152,
+    "eval_mar_100_Gloves": 0.3964,
+    "eval_mar_100_Goggles": 0.3615,
+    "eval_mar_100_Mask": 0.3871,
+    "eval_mar_large": 0.6949,
+    "eval_mar_medium": 0.3795,
+    "eval_mar_small": 0.224,
+    "eval_runtime": 7.7617,
+    "eval_samples_per_second": 19.326,
+    "eval_steps_per_second": 2.448
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "test_loss": 1.587841272354126,
+    "test_map": 0.3942,
+    "test_map_50": 0.7048,
+    "test_map_75": 0.3809,
+    "test_map_Coverall": 0.5521,
+    "test_map_Face_Shield": 0.575,
+    "test_map_Gloves": 0.3106,
+    "test_map_Goggles": 0.1884,
+    "test_map_Mask": 0.3449,
+    "test_map_large": 0.6081,
+    "test_map_medium": 0.2404,
+    "test_map_small": 0.422,
+    "test_mar_1": 0.3136,
+    "test_mar_10": 0.5138,
+    "test_mar_100": 0.5274,
+    "test_mar_100_Coverall": 0.6711,
+    "test_mar_100_Face_Shield": 0.7118,
+    "test_mar_100_Gloves": 0.4459,
+    "test_mar_100_Goggles": 0.3313,
+    "test_mar_100_Mask": 0.4769,
+    "test_mar_large": 0.7511,
+    "test_mar_medium": 0.4155,
+    "test_mar_small": 0.4541,
+    "test_runtime": 2.2158,
+    "test_samples_per_second": 13.088,
+    "test_steps_per_second": 1.805
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 100.0,
+    "total_flos": 1.643572308404736e+19,
+    "train_loss": 0.5523391162123635,
+    "train_runtime": 7636.311,
+    "train_samples_per_second": 11.131,
+    "train_steps_per_second": 1.401
+}