Update create_submission.py

create_submission.py

@@ -8,9 +8,10 @@ output_csv = "submission.csv"
 # Load frame-level results
 df = pd.read_csv(input_csv)
 
-# Extract base video ID (remove _f###.jpg)
+# Extract base video ID (remove _frame_### and file extension)
 def extract_video_id(filename):
-    return re.sub(r"_f\d+\.\w+$", "", filename)
+    # Remove patterns like _frame_1, _frame_2, etc, and optional extensions (.png, .jpg, etc)
+    return re.sub(r"_frame_\d+(?:\.\w+)?$", "", filename)
 
 df["video_id"] = df["file_name"].apply(extract_video_id)
 
@@ -32,4 +33,4 @@ video_scores = video_scores[["id", "pred", "score"]]
 
 # Save submission file
 video_scores.to_csv(output_csv, index=False)
-print(f"Saved submission file to {output_csv}")
+print(f"✅ Saved submission file to {output_csv}")