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-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/kri1hc22)
 # hustvl-yolos-small-finetuned-10k-cppe5-auto-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.7415
-- Map: 0.2892
-- Map 50: 0.5968
-- Map 75: 0.2545
-- Map Small: 0.1595
-- Map Medium: 0.2179
-- Map Large: 0.495
-- Mar 1: 0.2835
-- Mar 10: 0.4397
-- Mar 100: 0.4617
-- Mar Small: 0.2301
-- Mar Medium: 0.4069
-- Mar Large: 0.6909
-- Map Coverall: 0.5487
-- Mar 100 Coverall: 0.6842
-- Map Face Shield: 0.2878
-- Mar 100 Face Shield: 0.4494
-- Map Gloves: 0.1891
-- Mar 100 Gloves: 0.3924
-- Map Goggles: 0.1726
-- Mar 100 Goggles: 0.3815
-- Map Mask: 0.2478
-- Mar 100 Mask: 0.4009
+- Loss: 1.7709
+- Map: 0.298
+- Map 50: 0.6103
+- Map 75: 0.2583
+- Map Small: 0.1584
+- Map Medium: 0.2312
+- Map Large: 0.512
+- Mar 1: 0.2861
+- Mar 10: 0.4451
+- Mar 100: 0.4684
+- Mar Small: 0.2274
+- Mar Medium: 0.4152
+- Mar Large: 0.6937
+- Map Coverall: 0.5464
+- Mar 100 Coverall: 0.6914
+- Map Face Shield: 0.3047
+- Mar 100 Face Shield: 0.4671
+- Map Gloves: 0.2021
+- Mar 100 Gloves: 0.3987
+- Map Goggles: 0.1799
+- Mar 100 Goggles: 0.3723
+- Map Mask: 0.257
+- Mar 100 Mask: 0.4124
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.7708603143692017,
+    "eval_map": 0.298,
+    "eval_map_50": 0.6103,
+    "eval_map_75": 0.2583,
+    "eval_map_Coverall": 0.5464,
+    "eval_map_Face_Shield": 0.3047,
+    "eval_map_Gloves": 0.2021,
+    "eval_map_Goggles": 0.1799,
+    "eval_map_Mask": 0.257,
+    "eval_map_large": 0.512,
+    "eval_map_medium": 0.2312,
+    "eval_map_small": 0.1584,
+    "eval_mar_1": 0.2861,
+    "eval_mar_10": 0.4451,
+    "eval_mar_100": 0.4684,
+    "eval_mar_100_Coverall": 0.6914,
+    "eval_mar_100_Face_Shield": 0.4671,
+    "eval_mar_100_Gloves": 0.3987,
+    "eval_mar_100_Goggles": 0.3723,
+    "eval_mar_100_Mask": 0.4124,
+    "eval_mar_large": 0.6937,
+    "eval_mar_medium": 0.4152,
+    "eval_mar_small": 0.2274,
+    "eval_runtime": 7.4879,
+    "eval_samples_per_second": 20.032,
+    "eval_steps_per_second": 2.537,
+    "test_loss": 1.5367867946624756,
+    "test_map": 0.3974,
+    "test_map_50": 0.7169,
+    "test_map_75": 0.4036,
+    "test_map_Coverall": 0.5685,
+    "test_map_Face_Shield": 0.5919,
+    "test_map_Gloves": 0.3667,
+    "test_map_Goggles": 0.133,
+    "test_map_Mask": 0.3269,
+    "test_map_large": 0.5657,
+    "test_map_medium": 0.2449,
+    "test_map_small": 0.408,
+    "test_mar_1": 0.3145,
+    "test_mar_10": 0.5313,
+    "test_mar_100": 0.5483,
+    "test_mar_100_Coverall": 0.7267,
+    "test_mar_100_Face_Shield": 0.7412,
+    "test_mar_100_Gloves": 0.4787,
+    "test_mar_100_Goggles": 0.3219,
+    "test_mar_100_Mask": 0.4731,
+    "test_mar_large": 0.7141,
+    "test_mar_medium": 0.4036,
+    "test_mar_small": 0.4607,
+    "test_runtime": 2.0043,
+    "test_samples_per_second": 14.469,
+    "test_steps_per_second": 1.996,
+    "total_flos": 1.6194857431601019e+19,
+    "train_loss": 0.5626228197044302,
+    "train_runtime": 7619.966,
+    "train_samples_per_second": 11.155,
+    "train_steps_per_second": 1.404
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "eval_loss": 1.7708603143692017,
+    "eval_map": 0.298,
+    "eval_map_50": 0.6103,
+    "eval_map_75": 0.2583,
+    "eval_map_Coverall": 0.5464,
+    "eval_map_Face_Shield": 0.3047,
+    "eval_map_Gloves": 0.2021,
+    "eval_map_Goggles": 0.1799,
+    "eval_map_Mask": 0.257,
+    "eval_map_large": 0.512,
+    "eval_map_medium": 0.2312,
+    "eval_map_small": 0.1584,
+    "eval_mar_1": 0.2861,
+    "eval_mar_10": 0.4451,
+    "eval_mar_100": 0.4684,
+    "eval_mar_100_Coverall": 0.6914,
+    "eval_mar_100_Face_Shield": 0.4671,
+    "eval_mar_100_Gloves": 0.3987,
+    "eval_mar_100_Goggles": 0.3723,
+    "eval_mar_100_Mask": 0.4124,
+    "eval_mar_large": 0.6937,
+    "eval_mar_medium": 0.4152,
+    "eval_mar_small": 0.2274,
+    "eval_runtime": 7.4879,
+    "eval_samples_per_second": 20.032,
+    "eval_steps_per_second": 2.537
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "test_loss": 1.5367867946624756,
+    "test_map": 0.3974,
+    "test_map_50": 0.7169,
+    "test_map_75": 0.4036,
+    "test_map_Coverall": 0.5685,
+    "test_map_Face_Shield": 0.5919,
+    "test_map_Gloves": 0.3667,
+    "test_map_Goggles": 0.133,
+    "test_map_Mask": 0.3269,
+    "test_map_large": 0.5657,
+    "test_map_medium": 0.2449,
+    "test_map_small": 0.408,
+    "test_mar_1": 0.3145,
+    "test_mar_10": 0.5313,
+    "test_mar_100": 0.5483,
+    "test_mar_100_Coverall": 0.7267,
+    "test_mar_100_Face_Shield": 0.7412,
+    "test_mar_100_Gloves": 0.4787,
+    "test_mar_100_Goggles": 0.3219,
+    "test_mar_100_Mask": 0.4731,
+    "test_mar_large": 0.7141,
+    "test_mar_medium": 0.4036,
+    "test_mar_small": 0.4607,
+    "test_runtime": 2.0043,
+    "test_samples_per_second": 14.469,
+    "test_steps_per_second": 1.996
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 100.0,
+    "total_flos": 1.6194857431601019e+19,
+    "train_loss": 0.5626228197044302,
+    "train_runtime": 7619.966,
+    "train_samples_per_second": 11.155,
+    "train_steps_per_second": 1.404
+}