End of training

README.md

@@ -3,6 +3,8 @@ library_name: transformers
 license: apache-2.0
 base_model: PekingU/rtdetr_r50vd
 tags:
+- object-detection
+- vision
 - generated_from_trainer
 model-index:
 - name: suas-2025-rtdetr-finetuned-b16-lr1e-4
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # suas-2025-rtdetr-finetuned-b16-lr1e-4
 
-This model is a fine-tuned version of [PekingU/rtdetr_r50vd](https://huggingface.co/PekingU/rtdetr_r50vd) on an unknown dataset.
+This model is a fine-tuned version of [PekingU/rtdetr_r50vd](https://huggingface.co/PekingU/rtdetr_r50vd) on the mfly-auton/suas-2025-synthetic-data dataset.
 It achieves the following results on the evaluation set:
 - Loss: 5.5169
 - Map: 0.7975

all_results.json

@@ -1,54 +1,54 @@
 {
     "epoch": 20.0,
-    "test_loss": 4.6487956047058105,
-    "test_map": 0.8602,
-    "test_map_50": 0.9588,
-    "test_map_75": 0.9287,
-    "test_map_baseball-bat": 0.5145,
-    "test_map_basketball": 0.8925,
+    "test_loss": 4.647531032562256,
+    "test_map": 0.8603,
+    "test_map_50": 0.9592,
+    "test_map_75": 0.9285,
+    "test_map_baseball-bat": 0.513,
+    "test_map_basketball": 0.8923,
     "test_map_car": -1.0,
-    "test_map_football": 0.6313,
-    "test_map_human": 0.9186,
-    "test_map_large": 0.9699,
-    "test_map_luggage": 0.8921,
+    "test_map_football": 0.632,
+    "test_map_human": 0.9196,
+    "test_map_large": 0.971,
+    "test_map_luggage": 0.8911,
     "test_map_mattress": 0.9812,
-    "test_map_medium": 0.8498,
-    "test_map_motorcycle": 0.9013,
-    "test_map_skis": 0.9669,
-    "test_map_small": 0.702,
-    "test_map_snowboard": 0.9779,
-    "test_map_soccer-ball": 0.8238,
+    "test_map_medium": 0.8489,
+    "test_map_motorcycle": 0.9037,
+    "test_map_skis": 0.9644,
+    "test_map_small": 0.7019,
+    "test_map_snowboard": 0.9786,
+    "test_map_soccer-ball": 0.8239,
     "test_map_stop-sign": 0.9746,
-    "test_map_tennis-racket": 0.8141,
-    "test_map_umbrella": 0.9101,
-    "test_map_volleyball": 0.8435,
-    "test_mar_1": 0.8039,
+    "test_map_tennis-racket": 0.8156,
+    "test_map_umbrella": 0.9107,
+    "test_map_volleyball": 0.843,
+    "test_mar_1": 0.8038,
     "test_mar_10": 0.9143,
-    "test_mar_100": 0.9167,
-    "test_mar_100_baseball-bat": 0.7644,
-    "test_mar_100_basketball": 0.9259,
+    "test_mar_100": 0.9166,
+    "test_mar_100_baseball-bat": 0.7625,
+    "test_mar_100_basketball": 0.9257,
     "test_mar_100_car": -1.0,
-    "test_mar_100_football": 0.7329,
+    "test_mar_100_football": 0.7339,
     "test_mar_100_human": 0.9633,
-    "test_mar_100_luggage": 0.9482,
+    "test_mar_100_luggage": 0.9475,
     "test_mar_100_mattress": 0.994,
-    "test_mar_100_motorcycle": 0.9575,
+    "test_mar_100_motorcycle": 0.958,
     "test_mar_100_skis": 0.9935,
-    "test_mar_100_snowboard": 0.9889,
+    "test_mar_100_snowboard": 0.989,
     "test_mar_100_soccer-ball": 0.8611,
-    "test_mar_100_stop-sign": 0.9887,
+    "test_mar_100_stop-sign": 0.989,
     "test_mar_100_tennis-racket": 0.8852,
-    "test_mar_100_umbrella": 0.9526,
-    "test_mar_100_volleyball": 0.8783,
-    "test_mar_large": 0.9887,
-    "test_mar_medium": 0.9276,
-    "test_mar_small": 0.7594,
-    "test_runtime": 72.2106,
-    "test_samples_per_second": 20.773,
-    "test_steps_per_second": 0.651,
+    "test_mar_100_umbrella": 0.9524,
+    "test_mar_100_volleyball": 0.8775,
+    "test_mar_large": 0.9888,
+    "test_mar_medium": 0.9272,
+    "test_mar_small": 0.7585,
+    "test_runtime": 102.6229,
+    "test_samples_per_second": 14.617,
+    "test_steps_per_second": 0.916,
     "total_flos": 4.4130028142592e+19,
-    "train_loss": 5.691855750236337,
-    "train_runtime": 6795.0216,
-    "train_samples_per_second": 20.603,
-    "train_steps_per_second": 0.645
+    "train_loss": 0.0,
+    "train_runtime": 6.4113,
+    "train_samples_per_second": 10918.232,
+    "train_steps_per_second": 341.585
 }

test_results.json

@@ -1,49 +1,49 @@
 {
     "epoch": 20.0,
-    "test_loss": 4.6487956047058105,
-    "test_map": 0.8602,
-    "test_map_50": 0.9588,
-    "test_map_75": 0.9287,
-    "test_map_baseball-bat": 0.5145,
-    "test_map_basketball": 0.8925,
+    "test_loss": 4.647531032562256,
+    "test_map": 0.8603,
+    "test_map_50": 0.9592,
+    "test_map_75": 0.9285,
+    "test_map_baseball-bat": 0.513,
+    "test_map_basketball": 0.8923,
     "test_map_car": -1.0,
-    "test_map_football": 0.6313,
-    "test_map_human": 0.9186,
-    "test_map_large": 0.9699,
-    "test_map_luggage": 0.8921,
+    "test_map_football": 0.632,
+    "test_map_human": 0.9196,
+    "test_map_large": 0.971,
+    "test_map_luggage": 0.8911,
     "test_map_mattress": 0.9812,
-    "test_map_medium": 0.8498,
-    "test_map_motorcycle": 0.9013,
-    "test_map_skis": 0.9669,
-    "test_map_small": 0.702,
-    "test_map_snowboard": 0.9779,
-    "test_map_soccer-ball": 0.8238,
+    "test_map_medium": 0.8489,
+    "test_map_motorcycle": 0.9037,
+    "test_map_skis": 0.9644,
+    "test_map_small": 0.7019,
+    "test_map_snowboard": 0.9786,
+    "test_map_soccer-ball": 0.8239,
     "test_map_stop-sign": 0.9746,
-    "test_map_tennis-racket": 0.8141,
-    "test_map_umbrella": 0.9101,
-    "test_map_volleyball": 0.8435,
-    "test_mar_1": 0.8039,
+    "test_map_tennis-racket": 0.8156,
+    "test_map_umbrella": 0.9107,
+    "test_map_volleyball": 0.843,
+    "test_mar_1": 0.8038,
     "test_mar_10": 0.9143,
-    "test_mar_100": 0.9167,
-    "test_mar_100_baseball-bat": 0.7644,
-    "test_mar_100_basketball": 0.9259,
+    "test_mar_100": 0.9166,
+    "test_mar_100_baseball-bat": 0.7625,
+    "test_mar_100_basketball": 0.9257,
     "test_mar_100_car": -1.0,
-    "test_mar_100_football": 0.7329,
+    "test_mar_100_football": 0.7339,
     "test_mar_100_human": 0.9633,
-    "test_mar_100_luggage": 0.9482,
+    "test_mar_100_luggage": 0.9475,
     "test_mar_100_mattress": 0.994,
-    "test_mar_100_motorcycle": 0.9575,
+    "test_mar_100_motorcycle": 0.958,
     "test_mar_100_skis": 0.9935,
-    "test_mar_100_snowboard": 0.9889,
+    "test_mar_100_snowboard": 0.989,
     "test_mar_100_soccer-ball": 0.8611,
-    "test_mar_100_stop-sign": 0.9887,
+    "test_mar_100_stop-sign": 0.989,
     "test_mar_100_tennis-racket": 0.8852,
-    "test_mar_100_umbrella": 0.9526,
-    "test_mar_100_volleyball": 0.8783,
-    "test_mar_large": 0.9887,
-    "test_mar_medium": 0.9276,
-    "test_mar_small": 0.7594,
-    "test_runtime": 72.2106,
-    "test_samples_per_second": 20.773,
-    "test_steps_per_second": 0.651
+    "test_mar_100_umbrella": 0.9524,
+    "test_mar_100_volleyball": 0.8775,
+    "test_mar_large": 0.9888,
+    "test_mar_medium": 0.9272,
+    "test_mar_small": 0.7585,
+    "test_runtime": 102.6229,
+    "test_samples_per_second": 14.617,
+    "test_steps_per_second": 0.916
 }

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 20.0,
     "total_flos": 4.4130028142592e+19,
-    "train_loss": 5.691855750236337,
-    "train_runtime": 6795.0216,
-    "train_samples_per_second": 20.603,
-    "train_steps_per_second": 0.645
+    "train_loss": 0.0,
+    "train_runtime": 6.4113,
+    "train_samples_per_second": 10918.232,
+    "train_steps_per_second": 341.585
 }

trainer_state.json

@@ -1152,16 +1152,16 @@
       "epoch": 20.0,
       "step": 4380,
       "total_flos": 4.4130028142592e+19,
-      "train_loss": 5.691855750236337,
-      "train_runtime": 6795.0216,
-      "train_samples_per_second": 20.603,
-      "train_steps_per_second": 0.645
+      "train_loss": 0.0,
+      "train_runtime": 6.4113,
+      "train_samples_per_second": 10918.232,
+      "train_steps_per_second": 341.585
     }
   ],
   "logging_steps": 500,
-  "max_steps": 4380,
+  "max_steps": 2190,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 20,
+  "num_train_epochs": 10,
   "save_steps": 500,
   "stateful_callbacks": {
     "TrainerControl": {