kalpitbcontrails commited on
Commit
f625f4e
·
verified ·
1 Parent(s): be0a2fb

Update create_submission.py

Browse files
Files changed (1) hide show
  1. create_submission.py +4 -0
create_submission.py CHANGED
@@ -8,6 +8,10 @@ output_csv = "submission.csv"
8
  # Load frame-level results
9
  df = pd.read_csv(input_csv)
10
 
 
 
 
 
11
  # Extract base video ID (remove _frame_### and file extension)
12
  def extract_video_id(filename):
13
  # Remove patterns like _frame_1, _frame_2, etc, and optional extensions (.png, .jpg, etc)
 
8
  # Load frame-level results
9
  df = pd.read_csv(input_csv)
10
 
11
+ # ✅ Print the first few rows of results.csv
12
+ print("📄 Preview of results.csv:")
13
+ print(df.head(), "\n") # shows the first 5 rows
14
+
15
  # Extract base video ID (remove _frame_### and file extension)
16
  def extract_video_id(filename):
17
  # Remove patterns like _frame_1, _frame_2, etc, and optional extensions (.png, .jpg, etc)