Files
guitar_score/scripts/debug/dump_ocr.py

22 lines
762 B
Python

import cv2
import sys
from pathlib import Path
from youtube_tab_to_pdf import extract_frames, extract_unique_overlay, _extract_number_above_bars
from video_cv_tracker import TemporalTracker
video_path = Path("output/サカナクション/新宝島(エレキギターTAB) 難易度★★★ sakanaction shintakarajima.mp4")
frames = extract_frames(video_path, fps=2.0)
print(f"Extracted {len(frames)} frames.")
# Tracker extracts raw pages
tracker = TemporalTracker(diff_threshold=0.03)
for frame in frames:
tracker.process_frame(frame)
pages = tracker.get_unique_pages()
print(f"Tracker found {len(pages)} unique pages.")
for i, page in enumerate(pages):
m_num = _extract_number_above_bars(page)
print(f"Page {i:03d} OCR Measure Num: {m_num}")