Video converters enable transformations involving video files, particularly adding images to videos.
Overview¶
This notebook covers:
Image to Video: Add images to video files
Image to Video¶
AddImageVideoConverter¶
The AddImageVideoConverter adds an image overlay to a video file.
To use this converter you’ll need to install opencv which can be done with pip install pyrit[opencv]
import pathlib
from pyrit.prompt_converter import AddImageVideoConverter
from pyrit.setup import IN_MEMORY, initialize_pyrit_async
await initialize_pyrit_async(memory_db_type=IN_MEMORY) # type: ignore
input_video = str(pathlib.Path(".") / ".." / ".." / ".." / "assets" / "sample_video.mp4")
input_image = str(pathlib.Path(".") / ".." / ".." / ".." / "assets" / "pyrit_architecture.png")
video = AddImageVideoConverter(video_path=input_video)
converted_vid = await video.convert_async(prompt=input_image, input_type="image_path") # type: ignore
print(converted_vid)Found default environment files: ['./.pyrit/.env', './.pyrit/.env.local']
Loaded environment file: ./.pyrit/.env
Loaded environment file: ./.pyrit/.env.local
No new upgrade operations detected.
video_path: ./dbdata/prompt-memory-entries/videos/1779402988081284.mp4