Continuity / tests /create_assets.py
Gaurav vashistha
feat: Production deployment release
e0ea972
raw
history blame contribute delete
667 Bytes
import cv2
import numpy as np
import os
def make_video(filename, color):
height, width = 480, 640
# Use 'mp4v' for .mp4 containers
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
out = cv2.VideoWriter(filename, fourcc, 30.0, (width, height))
# Create a frame with the solid color
frame = np.zeros((height, width, 3), dtype=np.uint8)
frame[:] = color # BGR format
# Write 30 frames (1 second)
for _ in range(30):
out.write(frame)
out.release()
print(f"Created {filename}")
if __name__ == "__main__":
make_video('tests/scene_a.mp4', (255, 0, 0)) # Blue
make_video('tests/scene_c.mp4', (0, 0, 255)) # Red