End of training

README.md

@@ -1,6 +1,8 @@
 ---
 base_model: sbchoi/rtdetr_r50vd
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: sbchoi-rtdetr_r50vd-finetuned-10k-cppe5
@@ -13,7 +15,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/koo968ri)
 # sbchoi-rtdetr_r50vd-finetuned-10k-cppe5
 
-This model is a fine-tuned version of [sbchoi/rtdetr_r50vd](https://huggingface.co/sbchoi/rtdetr_r50vd) on an unknown dataset.
+This model is a fine-tuned version of [sbchoi/rtdetr_r50vd](https://huggingface.co/sbchoi/rtdetr_r50vd) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
 - Loss: 5.4784
 - Map: 0.2239

all_results.json

@@ -0,0 +1,34 @@
+{
+    "epoch": 100.0,
+    "test_loss": 4.624134063720703,
+    "test_map": 0.2868,
+    "test_map_50": 0.5176,
+    "test_map_75": 0.2575,
+    "test_map_Coverall": 0.3316,
+    "test_map_Face_Shield": 0.2115,
+    "test_map_Gloves": 0.3246,
+    "test_map_Goggles": 0.0636,
+    "test_map_Mask": 0.5028,
+    "test_map_large": 0.4459,
+    "test_map_medium": 0.2024,
+    "test_map_small": 0.0708,
+    "test_mar_1": 0.226,
+    "test_mar_10": 0.3749,
+    "test_mar_100": 0.388,
+    "test_mar_100_Coverall": 0.5359,
+    "test_mar_100_Face_Shield": 0.2882,
+    "test_mar_100_Gloves": 0.4627,
+    "test_mar_100_Goggles": 0.0828,
+    "test_mar_100_Mask": 0.5706,
+    "test_mar_large": 0.5355,
+    "test_mar_medium": 0.2626,
+    "test_mar_small": 0.0875,
+    "test_runtime": 3.0589,
+    "test_samples_per_second": 9.48,
+    "test_steps_per_second": 1.308,
+    "total_flos": 2.6781957255168e+19,
+    "train_loss": 5.885154977958893,
+    "train_runtime": 8264.1468,
+    "train_samples_per_second": 10.285,
+    "train_steps_per_second": 1.295
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 100.0,
+    "test_loss": 4.624134063720703,
+    "test_map": 0.2868,
+    "test_map_50": 0.5176,
+    "test_map_75": 0.2575,
+    "test_map_Coverall": 0.3316,
+    "test_map_Face_Shield": 0.2115,
+    "test_map_Gloves": 0.3246,
+    "test_map_Goggles": 0.0636,
+    "test_map_Mask": 0.5028,
+    "test_map_large": 0.4459,
+    "test_map_medium": 0.2024,
+    "test_map_small": 0.0708,
+    "test_mar_1": 0.226,
+    "test_mar_10": 0.3749,
+    "test_mar_100": 0.388,
+    "test_mar_100_Coverall": 0.5359,
+    "test_mar_100_Face_Shield": 0.2882,
+    "test_mar_100_Gloves": 0.4627,
+    "test_mar_100_Goggles": 0.0828,
+    "test_mar_100_Mask": 0.5706,
+    "test_mar_large": 0.5355,
+    "test_mar_medium": 0.2626,
+    "test_mar_small": 0.0875,
+    "test_runtime": 3.0589,
+    "test_samples_per_second": 9.48,
+    "test_steps_per_second": 1.308
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 100.0,
+    "total_flos": 2.6781957255168e+19,
+    "train_loss": 5.885154977958893,
+    "train_runtime": 8264.1468,
+    "train_samples_per_second": 10.285,
+    "train_steps_per_second": 1.295
+}