Spaces:
Paused
Paused
File size: 1,839 Bytes
8f0cb79 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import cv2
import os
import hashlib
import numpy as np
from moviepy import ImageSequenceClip
def create_video_from_images(image_folder="images", output_video="videos/output_video.mp4", fps=30):
"""Create MP4 with lossless encoding using MoviePy"""
os.makedirs(os.path.dirname(output_video), exist_ok=True)
# Get sorted list of images
image_files = sorted([os.path.join(image_folder, f)
for f in os.listdir(image_folder) if f.endswith('.png')],
key=lambda x: int(os.path.basename(x).split('.')[0]))
# Create video with lossless settings
clip = ImageSequenceClip(image_files, fps=fps)
# FFmpeg parameters for lossless encoding
clip.write_videofile(
output_video,
codec='libx264rgb',
preset='ultrafast',
ffmpeg_params=[
'-crf', '0', # Lossless
'-color_range', '2', # Full color range
'-pix_fmt', 'bgr24' # Match OpenCV's BGR format
]
)
def extract_frames_from_video(video_path="videos/output_video.mp4", output_folder="extracted_frames"):
"""Extract frames with OpenCV and hash verification"""
os.makedirs(output_folder, exist_ok=True)
cap = cv2.VideoCapture(video_path)
frame_count = 0
while True:
success, frame = cap.read()
if not success:
break
output_path = os.path.join(output_folder, f"{frame_count}.png")
cv2.imwrite(output_path, frame, [cv2.IMWRITE_PNG_COMPRESSION, 0])
with open(output_path, 'rb') as f:
current_hash = hashlib.sha256(f.read()).hexdigest()
print(f"Frame {frame_count:04d} hash: {current_hash}")
frame_count += 1
cap.release()
|