Update create_submission.py
Browse files- 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)
|