def compute_reference_embedding()

in video_processing/reference_video_similarity.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

def compute_reference_embedding(ref_path, model, preprocessor, device, dtype):
    """
    Computes the embedding for a reference file (image or video).
    """
    video_extensions = (".mp4", ".avi", ".mov", ".mkv")
    if ref_path.lower().endswith(video_extensions):
        frames = get_frames(ref_path)
        frames = next(iter(frames))
        frames = [frame.to_image() for frame in frames]
        return compute_video_embedding(frames, model, preprocessor, device, dtype)
    else:
        return compute_image_embedding(ref_path, model, preprocessor, device, dtype)