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
@@ -14,7 +16,7 @@ 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/u2o06hbj)
 # hustvl-yolos-small-finetuned-10k-cppe5
 
-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.4674
 - Map: 0.3363

all_results.json

@@ -1,8 +1,34 @@
 {
     "epoch": 100.0,
-    "total_flos": 5.80084344142848e+16,
-    "train_loss": 1.202735595703125,
-    "train_runtime": 44.9423,
-    "train_samples_per_second": 6.675,
-    "train_steps_per_second": 2.225
+    "test_loss": 1.4859906435012817,
+    "test_map": 0.3129,
+    "test_map_50": 0.5908,
+    "test_map_75": 0.3127,
+    "test_map_Coverall": 0.4366,
+    "test_map_Face_Shield": 0.4978,
+    "test_map_Gloves": 0.2774,
+    "test_map_Goggles": 0.0678,
+    "test_map_Mask": 0.2847,
+    "test_map_large": 0.4646,
+    "test_map_medium": 0.1913,
+    "test_map_small": 0.3656,
+    "test_mar_1": 0.2966,
+    "test_mar_10": 0.4818,
+    "test_mar_100": 0.4962,
+    "test_mar_100_Coverall": 0.7051,
+    "test_mar_100_Face_Shield": 0.6706,
+    "test_mar_100_Gloves": 0.4186,
+    "test_mar_100_Goggles": 0.2552,
+    "test_mar_100_Mask": 0.4314,
+    "test_mar_large": 0.6605,
+    "test_mar_medium": 0.3512,
+    "test_mar_small": 0.4247,
+    "test_runtime": 1.9456,
+    "test_samples_per_second": 14.906,
+    "test_steps_per_second": 2.056,
+    "total_flos": 1.619523370252695e+19,
+    "train_loss": 0.5915974802391551,
+    "train_runtime": 7701.1118,
+    "train_samples_per_second": 11.037,
+    "train_steps_per_second": 1.389
 }

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "test_loss": 1.4859906435012817,
+    "test_map": 0.3129,
+    "test_map_50": 0.5908,
+    "test_map_75": 0.3127,
+    "test_map_Coverall": 0.4366,
+    "test_map_Face_Shield": 0.4978,
+    "test_map_Gloves": 0.2774,
+    "test_map_Goggles": 0.0678,
+    "test_map_Mask": 0.2847,
+    "test_map_large": 0.4646,
+    "test_map_medium": 0.1913,
+    "test_map_small": 0.3656,
+    "test_mar_1": 0.2966,
+    "test_mar_10": 0.4818,
+    "test_mar_100": 0.4962,
+    "test_mar_100_Coverall": 0.7051,
+    "test_mar_100_Face_Shield": 0.6706,
+    "test_mar_100_Gloves": 0.4186,
+    "test_mar_100_Goggles": 0.2552,
+    "test_mar_100_Mask": 0.4314,
+    "test_mar_large": 0.6605,
+    "test_mar_medium": 0.3512,
+    "test_mar_small": 0.4247,
+    "test_runtime": 1.9456,
+    "test_samples_per_second": 14.906,
+    "test_steps_per_second": 2.056
+}

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 100.0,
-    "total_flos": 5.80084344142848e+16,
-    "train_loss": 1.202735595703125,
-    "train_runtime": 44.9423,
-    "train_samples_per_second": 6.675,
-    "train_steps_per_second": 2.225
+    "total_flos": 1.619523370252695e+19,
+    "train_loss": 0.5915974802391551,
+    "train_runtime": 7701.1118,
+    "train_samples_per_second": 11.037,
+    "train_steps_per_second": 1.389
 }