在PyQt5 GUI中使用多线程保存动画时的画质问题

问题背景 #

为优化用户体验，我在主线程完成动画帧绘制，将动画保存操作放到子线程执行以同步更新进度条，但发现子线程保存后的动画画质明显低于主线程直接保存的效果。

现有代码 #

CameraWorker子线程类

class CameraWorker(QObject):
    finished = pyqtSignal()

    def __init__(self, cam=None, dir=None, fps=None):
        QObject.__init__(self)
        self._cam = cam
        self._dir = dir
        self._fps = fps

    def gifsave(self):
        ani = self._cam.animate(blit=True)
        ani.save(self._dir, dpi=150, writer=PillowWriter(fps=self._fps, bitrate=30000))
        self.finished.emit()

主线程绘图与线程调度代码

def draw_phvideo(self):            
        fig = plt.figure()
        
        camera = Camera(fig)

        for targetTime in range_time:
            """ 帧绘制逻辑 """
            camera.snap()
        self.prgBr.setValue(0)
        self.prgBr.setMaximum(0)
        
        # 初始化子线程
        self.thrd_cam = QThread()
        self.wrkr_cam = CameraWorker(camera, fileName[0], fps)
        self.wrkr_cam.moveToThread(self.thrd_cam)

        # 绑定线程信号
        self.thrd_cam.started.connect(self.wrkr_cam.gifsave)
        self.wrkr_cam.finished.connect(self.done_cam)

        # 启动子线程
        self.thrd_cam.start()
        
def done_cam(self):
    print("multi thread done")
    self.prgBr.setMaximum(1)
    self.thrd_cam.quit()
    self.thrd_cam.wait()
    self.wrkr_cam.deleteLater()
    self.thrd_cam.deleteLater()

问题原因分析 #

1. 绘图上下文绑定问题：Camera对象是在主线程的绘图上下文创建的，子线程中调用camera.animate()时，无法正确继承主线程的高分辨率渲染配置，导致动画帧的尺寸或画质被降级。
2. blit=True的线程兼容性：blit=True依赖主线程的绘图缓冲区更新机制，子线程中使用会导致渲染不完整，间接拉低画质。

解决方案 #

方案1：主线程预生成动画对象，子线程仅执行保存

将animate()的调用移到主线程，把生成好的ani对象传给子线程，彻底避免子线程操作主线程的绘图上下文：

修改CameraWorker类：

class CameraWorker(QObject):
    finished = pyqtSignal()

    def __init__(self, ani=None, save_path=None):
        QObject.__init__(self)
        self._ani = ani
        self._save_path = save_path

    def gifsave(self):
        # 子线程仅执行保存操作
        self._ani.save(self._save_path, dpi=150, writer=PillowWriter(fps=self._ani._fps, bitrate=30000))
        self.finished.emit()

修改主线程绘图代码：

def draw_phvideo(self):            
        # 创建figure时直接指定dpi，确保渲染基础分辨率
        fig = plt.figure(dpi=150)
        
        camera = Camera(fig)

        for targetTime in range_time:
            """ 帧绘制逻辑 """
            camera.snap()
        
        # 主线程预生成动画对象
        ani = camera.animate(blit=True)
        
        self.prgBr.setValue(0)
        self.prgBr.setMaximum(0)
        
        # 初始化子线程，传入预生成的ani对象
        self.thrd_cam = QThread()
        self.wrkr_cam = CameraWorker(ani, fileName[0])
        self.wrkr_cam.moveToThread(self.thrd_cam)

        # 绑定线程信号
        self.thrd_cam.started.connect(self.wrkr_cam.gifsave)
        self.wrkr_cam.finished.connect(self.done_cam)

        # 启动子线程
        self.thrd_cam.start()

方案2：优化PillowWriter参数提升画质

如果方案1无法完全解决，可调整PillowWriter的参数进一步提升画质：

# 提高bitrate或指定PIL的保存质量
from matplotlib.animation import PillowWriter

# 方式1：提高比特率
writer = PillowWriter(fps=self._fps, bitrate=50000)
# 方式2：直接设置PIL的保存质量参数
writer = PillowWriter(fps=self._fps, save_kwargs={'quality': 95})
ani.save(self._save_path, dpi=150, writer=writer)

方案3：分离渲染与合成流程

如果上述方案仍有问题，可在主线程完成所有帧的渲染并导出为临时图像序列，子线程仅负责将图像序列合成GIF，彻底分离绘图与保存的上下文。

内容的提问来源于stack exchange，提问作者onlyhappiness