End of training

README.md

@@ -1,6 +1,8 @@
 ---
 base_model: jozhang97/deta-resnet-50
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: jozhang97-deta-resnet-50-finetuned-10k-cppe5-auto-pad
@@ -13,31 +15,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/pw3c0sr5)
 # jozhang97-deta-resnet-50-finetuned-10k-cppe5-auto-pad
 
-This model is a fine-tuned version of [jozhang97/deta-resnet-50](https://huggingface.co/jozhang97/deta-resnet-50) on an unknown dataset.
+This model is a fine-tuned version of [jozhang97/deta-resnet-50](https://huggingface.co/jozhang97/deta-resnet-50) on the cppe-5 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.6173
-- Map: 0.1965
-- Map 50: 0.4335
-- Map 75: 0.1626
-- Map Small: 0.0855
-- Map Medium: 0.1284
-- Map Large: 0.3222
-- Mar 1: 0.2092
-- Mar 10: 0.3167
-- Mar 100: 0.3327
-- Mar Small: 0.1181
-- Mar Medium: 0.2784
-- Mar Large: 0.5071
-- Map Coverall: 0.4181
-- Mar 100 Coverall: 0.5347
-- Map Face Shield: 0.227
-- Mar 100 Face Shield: 0.3633
-- Map Gloves: 0.0765
-- Mar 100 Gloves: 0.2134
-- Map Goggles: 0.1245
-- Mar 100 Goggles: 0.2523
-- Map Mask: 0.1363
-- Mar 100 Mask: 0.3
+- Loss: 1.6197
+- Map: 0.1981
+- Map 50: 0.434
+- Map 75: 0.1623
+- Map Small: 0.0801
+- Map Medium: 0.1333
+- Map Large: 0.3217
+- Mar 1: 0.2118
+- Mar 10: 0.3156
+- Mar 100: 0.3333
+- Mar Small: 0.1156
+- Mar Medium: 0.2789
+- Mar Large: 0.5084
+- Map Coverall: 0.4266
+- Mar 100 Coverall: 0.5446
+- Map Face Shield: 0.2284
+- Mar 100 Face Shield: 0.3658
+- Map Gloves: 0.0771
+- Mar 100 Gloves: 0.2098
+- Map Goggles: 0.1232
+- Mar 100 Goggles: 0.2477
+- Map Mask: 0.1352
+- Mar 100 Mask: 0.2987
 
 ## Model description
 

all_results.json

@@ -0,0 +1,60 @@
+{
+    "epoch": 99.53051643192488,
+    "eval_loss": 1.6197102069854736,
+    "eval_map": 0.1981,
+    "eval_map_50": 0.434,
+    "eval_map_75": 0.1623,
+    "eval_map_Coverall": 0.4266,
+    "eval_map_Face_Shield": 0.2284,
+    "eval_map_Gloves": 0.0771,
+    "eval_map_Goggles": 0.1232,
+    "eval_map_Mask": 0.1352,
+    "eval_map_large": 0.3217,
+    "eval_map_medium": 0.1333,
+    "eval_map_small": 0.0801,
+    "eval_mar_1": 0.2118,
+    "eval_mar_10": 0.3156,
+    "eval_mar_100": 0.3333,
+    "eval_mar_100_Coverall": 0.5446,
+    "eval_mar_100_Face_Shield": 0.3658,
+    "eval_mar_100_Gloves": 0.2098,
+    "eval_mar_100_Goggles": 0.2477,
+    "eval_mar_100_Mask": 0.2987,
+    "eval_mar_large": 0.5084,
+    "eval_mar_medium": 0.2789,
+    "eval_mar_small": 0.1156,
+    "eval_runtime": 21.4083,
+    "eval_samples_per_second": 7.007,
+    "eval_steps_per_second": 0.888,
+    "test_loss": 1.6902095079421997,
+    "test_map": 0.264,
+    "test_map_50": 0.559,
+    "test_map_75": 0.2055,
+    "test_map_Coverall": 0.4752,
+    "test_map_Face_Shield": 0.4395,
+    "test_map_Gloves": 0.0822,
+    "test_map_Goggles": 0.1197,
+    "test_map_Mask": 0.2037,
+    "test_map_large": 0.3582,
+    "test_map_medium": 0.1795,
+    "test_map_small": 0.2979,
+    "test_mar_1": 0.2365,
+    "test_mar_10": 0.3811,
+    "test_mar_100": 0.4012,
+    "test_mar_100_Coverall": 0.5689,
+    "test_mar_100_Face_Shield": 0.5882,
+    "test_mar_100_Gloves": 0.2295,
+    "test_mar_100_Goggles": 0.2656,
+    "test_mar_100_Mask": 0.3538,
+    "test_mar_large": 0.5382,
+    "test_mar_medium": 0.3294,
+    "test_mar_small": 0.3411,
+    "test_runtime": 4.6886,
+    "test_samples_per_second": 6.185,
+    "test_steps_per_second": 0.853,
+    "total_flos": 2.4566240955078824e+19,
+    "train_loss": 1.7565026848271208,
+    "train_runtime": 21477.6558,
+    "train_samples_per_second": 3.958,
+    "train_steps_per_second": 0.494
+}

eval_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 99.53051643192488,
+    "eval_loss": 1.6197102069854736,
+    "eval_map": 0.1981,
+    "eval_map_50": 0.434,
+    "eval_map_75": 0.1623,
+    "eval_map_Coverall": 0.4266,
+    "eval_map_Face_Shield": 0.2284,
+    "eval_map_Gloves": 0.0771,
+    "eval_map_Goggles": 0.1232,
+    "eval_map_Mask": 0.1352,
+    "eval_map_large": 0.3217,
+    "eval_map_medium": 0.1333,
+    "eval_map_small": 0.0801,
+    "eval_mar_1": 0.2118,
+    "eval_mar_10": 0.3156,
+    "eval_mar_100": 0.3333,
+    "eval_mar_100_Coverall": 0.5446,
+    "eval_mar_100_Face_Shield": 0.3658,
+    "eval_mar_100_Gloves": 0.2098,
+    "eval_mar_100_Goggles": 0.2477,
+    "eval_mar_100_Mask": 0.2987,
+    "eval_mar_large": 0.5084,
+    "eval_mar_medium": 0.2789,
+    "eval_mar_small": 0.1156,
+    "eval_runtime": 21.4083,
+    "eval_samples_per_second": 7.007,
+    "eval_steps_per_second": 0.888
+}

test_results.json

@@ -0,0 +1,29 @@
+{
+    "epoch": 99.53051643192488,
+    "test_loss": 1.6902095079421997,
+    "test_map": 0.264,
+    "test_map_50": 0.559,
+    "test_map_75": 0.2055,
+    "test_map_Coverall": 0.4752,
+    "test_map_Face_Shield": 0.4395,
+    "test_map_Gloves": 0.0822,
+    "test_map_Goggles": 0.1197,
+    "test_map_Mask": 0.2037,
+    "test_map_large": 0.3582,
+    "test_map_medium": 0.1795,
+    "test_map_small": 0.2979,
+    "test_mar_1": 0.2365,
+    "test_mar_10": 0.3811,
+    "test_mar_100": 0.4012,
+    "test_mar_100_Coverall": 0.5689,
+    "test_mar_100_Face_Shield": 0.5882,
+    "test_mar_100_Gloves": 0.2295,
+    "test_mar_100_Goggles": 0.2656,
+    "test_mar_100_Mask": 0.3538,
+    "test_mar_large": 0.5382,
+    "test_mar_medium": 0.3294,
+    "test_mar_small": 0.3411,
+    "test_runtime": 4.6886,
+    "test_samples_per_second": 6.185,
+    "test_steps_per_second": 0.853
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 99.53051643192488,
+    "total_flos": 2.4566240955078824e+19,
+    "train_loss": 1.7565026848271208,
+    "train_runtime": 21477.6558,
+    "train_samples_per_second": 3.958,
+    "train_steps_per_second": 0.494
+}