Function evaluate_video_file

evaluate.py:94–168

Evaluate a video file using the provided model. Args: model: The model to use for evaluation. video_path (str): Path to the video file. transcript_path (str): Path to the transcript file. description_path (str): Path to the description file. target_f

(model, video_path, transcript_path, description_path, target_fps=None, output_folder=None)

def evaluate_video_file(model, video_path, transcript_path, description_path, target_fps=None, output_folder=None):
    """
    Evaluate a video file using the provided model.

    Args:
        model: The model to use for evaluation.
        video_path (str): Path to the video file.
        transcript_path (str): Path to the transcript file.
        description_path (str): Path to the description file.
        target_fps (int, optional): Target frames per second for video processing.
        output_folder (str, optional): Directory to store output files.

    Returns:
        Dict or None: Evaluation results if successful, None if file format unsupported.
    """
    if not video_path.endswith(('.mp4', '.mkv')):
        print(f"Skipping {video_path}: Unsupported file format for video evaluation.")
        return None

    moviepy_temp_dir = os.path.join(output_folder, "moviepy_temp")

    # Chunking
    num_chunks = 10
    with VideoFileClip(video_path) as clip:
        duration = clip.duration
        chunk_duration = duration / num_chunks
        results = []
        
        # Create a temporary directory in the output_folder
        temp_dir_parent = output_folder or os.getcwd()
        with tempfile.TemporaryDirectory(dir=temp_dir_parent) as temp_dir:
            for i in range(10):
                start = i * chunk_duration
                end = min(start + chunk_duration, duration)
                chunk = clip.subclipped(start, end)
                chunk_path = os.path.join(temp_dir, f"chunk_{i+1}.mp4")
                # Explicitly set the temp_audiofile path with matching codec
                temp_audiofile = os.path.join(moviepy_temp_dir, f"temp_audio_chunk_{i+1}.m4a")
                chunk.write_videofile(
                    chunk_path,
                    codec="libx264",
                    audio_codec="aac",
                    temp_audiofile=temp_audiofile,
                    audio_bitrate="192k",
                    preset="ultrafast",  # Speed up encoding
                    logger=None
                )
                # Create processed videos folder inside output_folder
                processed_videos_dir = os.path.join(output_folder, "processed_videos")
                save_path = os.path.join(processed_videos_dir, f"processed_chunk_{i+1}.mp4")
                result = evaluate_video_chunk_new(
                    model,
                    chunk_path,
                    transcript_path,
                    description_path,
                    target_fps=target_fps,
                    save_processed_video=save_path
                )