Python多线程在音视频处理中的应用 Python多线程媒体处理优化

import threading
import subprocess
import os

def extract_thumbnail(video_path, output_path):
    cmd = [
        'ffmpeg', '-i', video_path,
        '-ss', '00:00:05', '-vframes', '1',
        output_path
    ]
    subprocess.run(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
    print(f"已生成缩略图: {output_path}")

# 多个视频并发处理
videos = [('video1.mp4', 'thumb1.jpg'), ('video2.mp4', 'thumb2.jpg')]

threads = []
for video, thumb in videos:
    if os.path.exists(video):
        t = threading.Thread(target=extract_thumbnail, args=(video, thumb))
        t.start()
        threads.append(t)

for t in threads:
    t.join()

from threading import Thread
import queue
import subprocess

def worker(q):
    while True:
        item = q.get()
        if item is None:
            break
        video, output = item
        subprocess.run([
            'ffmpeg', '-i', video, '-t', '10', output
        ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
        print(f"完成: {output}")
        q.task_done()

# 创建任务队列
q = queue.Queue()
num_worker_threads = 4

# 启动工作线程
threads = []
for _ in range(num_worker_threads):
    t = Thread(target=worker, args=(q,))
    t.start()
    threads.append(t)

# 添加任务
for i in range(10):
    q.put((f"input_{i}.mp4", f"clip_{i}.mp4"))

# 等待任务完成
q.join()

# 停止线程
for _ in range(num_worker_threads):
    q.put(None)
for t in threads:
    t.join()

from multiprocessing import Pool
import cv2

def process_video(filepath):
    cap = cv2.VideoCapture(filepath)
    frame_count = 0
    while cap.isOpened():
        ret, frame = cap.read()
        if not ret:
            break
        # 模拟处理
        frame_count += 1
    cap.release()
    return f"{filepath}: 处理 {frame_count} 帧"

if __name__ == '__main__':
    with Pool(4) as pool:
        results = pool.map(process_video, ['video1.mp4', 'video2.mp4'])
        print(results)