如何释放Python中Mediapipe Pose模型占用的内存？

问题核心 #

Mediapipe Pose的Python对象只是底层C资源的包装器，del和gc.collect()只能回收Python层的内存，无法触发C层资源的释放，导致内存泄漏；同一模型处理不同分辨率视频报错，是因为分割平滑计算器依赖历史帧尺寸，分辨率变化后断言条件不满足。

可行解决方案 #

• 显式调用资源释放方法：先调用模型的close()或release()方法（不同版本命名可能有差异），再删除对象并回收垃圾，确保C++层资源被释放：

poseObject.close()  # 优先调用此方法释放底层资源
del poseObject
gc.collect()

• 使用上下文管理器自动管理：如果Pose类支持上下文协议，用with语句创建模型，离开代码块时会自动释放所有资源，无需手动执行删除操作：

import mediapipe as mp

with mp.solutions.pose.Pose(
    static_image_mode=False,
    model_complexity=1,
    smooth_landmarks=True
) as pose:
    # 处理当前分辨率视频帧的逻辑
# 离开with块后，底层资源自动释放

• 子进程隔离内存：如果前两种方法无效，用多进程处理不同分辨率的任务。每个子进程独立运行，进程结束后系统会强制回收所有内存，彻底避免泄漏：

import multiprocessing
import mediapipe as mp

def handle_video(resolution, video_path):
    # 子进程内创建模型并处理
    with mp.solutions.pose.Pose(...) as pose:
        # 按当前分辨率处理视频的逻辑

# 遍历所有需要处理的分辨率，启动子进程
resolutions = [(1920, 1080), (1280, 720)]  # 示例分辨率列表
for res in resolutions:
    p = multiprocessing.Process(target=handle_video, args=(res, "your_video_path.mp4"))
    p.start()
    p.join()

• 升级Mediapipe版本：旧版本可能存在已知的内存泄漏bug，升级到最新稳定版大概率能解决问题：

pip install --upgrade mediapipe

内容的提问来源于stack exchange，提问作者Aras Umut Erarslan