Este script de Python utiliza OpenCV para extraer fotogramas específicos de un video. Los segundos de los fotogramas a extraer se especifican en un archivo CSV en la columna Frame
Este script requiere las bibliotecas de Python OpenCV, pandas y os. Puedes instalar estas dependencias con pip, el administrador de paquetes de Python, utilizando el archivo requirements.txt
incluido. Abre una terminal y ejecuta el siguiente comando:
pip install -r requirements.txt
El script define una función extract_frames
que toma tres argumentos:
video_path
: la ruta al archivo de video del que se extraerán los fotogramas.csv_path
: la ruta a un archivo CSV que contiene los segundos de los fotogramas a extraer. El archivo CSV debe tener una columna llamada 'Frame' que contiene los segundos.output_folder
: la ruta a la carpeta donde se guardarán los fotogramas extraídos. Si la carpeta no existe, se creará.size
(opcional): el tamaño deseado para el ancho y alto de las imágenes exportadas. Si se proporciona, todas las imágenes se redimensionarán a este tamaño. Si no se proporciona, las imágenes se guardarán con su tamaño original después de hacerlas cuadradas.
La función extract_frames
abre el video, lee el archivo CSV y extrae el fotograma correspondiente a cada segundo especificado en el CSV. Cada fotograma se guarda como una imagen PNG en la carpeta de salida.
extract_frames('video.mp4', 'data.csv', 'frames')
En este ejemplo, el script extraerá los fotogramas especificados en 'data.csv' del video 'video.mp4' y guardará las imágenes en la carpeta 'frames'.