23 lines
789 B
Python
23 lines
789 B
Python
import cv2
|
|
from pathlib import Path
|
|
from youtube_tab_to_pdf import extract_frames, detect_pattern, extract_unique_scroll, extract_unique_overlay, generate_pdf
|
|
|
|
video_path = Path("output/サカナクション／新宝島(エレキギターTAB) 難易度★★★ sakanaction shintakarajima.mp4")
|
|
output_pdf = Path("output/shintakarajima_perfect.pdf")
|
|
|
|
print("1. Extracting frames at 2fps...")
|
|
frames = extract_frames(video_path, fps=2.0)
|
|
|
|
pattern = detect_pattern(frames)
|
|
print(f"2. Detected Pattern: {pattern}")
|
|
|
|
if pattern == "overlay":
|
|
final_chunks = extract_unique_overlay(frames)
|
|
else:
|
|
final_chunks = extract_unique_scroll(frames)
|
|
|
|
print(f"3. Generating PDF with {len(final_chunks)} chunks...")
|
|
generate_pdf(final_chunks, output_pdf)
|
|
|
|
print(f"Done! PDF saved to {output_pdf}")
|