File size: 1,138 Bytes
a605a60 |
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 |
from transformers import pipeline
import cv2
import numpy as np
import os
# Initialize a text-to-image generation pipeline (for example purposes)
generator = pipeline('text-to-image', model='CompVis/stable-diffusion-v1-4')
# List of prompts for each frame
prompts = [
"A spaceship in space, vibrant colors",
"A spaceship flying past planets",
"A spaceship approaching a distant galaxy",
"A spaceship landing on an alien planet"
]
# Folder to save frames
os.makedirs('frames', exist_ok=True)
# Generate images for each prompt
for idx, prompt in enumerate(prompts):
image = generator(prompt)[0]['image']
image.save(f'frames/frame_{idx:03d}.png')
# Combine frames into a video
frame_files = [f'frames/frame_{i:03d}.png' for i in range(len(prompts))]
frame_width, frame_height = cv2.imread(frame_files[0]).shape[1], cv2.imread(frame_files[0]).shape[0]
video_writer = cv2.VideoWriter('space_video.mp4', cv2.VideoWriter_fourcc(*'mp4v'), 1, (frame_width, frame_height))
for file in frame_files:
img = cv2.imread(file)
video_writer.write(img)
video_writer.release()
print("Video created: space_video.mp4") |