树莓派Zero新系统与相机模块3：相机库兼容性问题求助

核心问题分析 #

旧代码依赖的imutils.video.pivideostream.PiVideoStream基于已废弃的picamera库，而新版树莓派系统默认使用picamera2，两者API完全不兼容，软链接或强行安装旧库无法解决根本问题，需要重写相机流逻辑。

解决方案步骤 #

1. 安装必要依赖 #

先确保系统和依赖库为最新版本：

sudo apt update && sudo apt upgrade -y
sudo apt install python3-picamera2 -y
pip install opencv-python numpy

2. 重写VideoCamera类（替代原PiVideoStream实现） #

直接基于picamera2和线程化读取，实现与原代码功能一致的相机流类：

import cv2
import time
import numpy as np
import threading
from picamera2 import Picamera2

class VideoCamera(object):
    def __init__(self, flip=False):
        self.flip = flip
        # 初始化Picamera2并配置参数
        self.picam2 = Picamera2()
        # 树莓派Zero建议用低分辨率提升流畅度，可按需调整
        camera_config = self.picam2.create_preview_configuration(
            main={"size": (640, 480), "format": "RGB888"}
        )
        self.picam2.configure(camera_config)
        self.picam2.start()
        time.sleep(2.0)  # 相机预热时间

        # 线程化读取帧，避免阻塞主线程
        self.running = True
        self.current_frame = None
        self.frame_lock = threading.Lock()
        self.read_thread = threading.Thread(target=self._update_frame, daemon=True)
        self.read_thread.start()

    def _update_frame(self):
        # 后台线程持续读取相机帧
        while self.running:
            frame = self.picam2.capture_array()
            with self.frame_lock:
                self.current_frame = frame
            time.sleep(0.01)  # 控制帧率，平衡性能与流畅度

    def __del__(self):
        # 资源释放逻辑
        self.running = False
        self.read_thread.join()
        self.picam2.stop()

    def flip_if_needed(self, frame):
        if self.flip:
            return np.flip(frame, 0)
        return frame

    def get_frame(self):
        # 获取编码后的JPEG帧，与原接口保持一致
        with self.frame_lock:
            # 避免帧未初始化时出错
            frame = self.current_frame.copy() if self.current_frame is not None else np.zeros((480, 640, 3), dtype=np.uint8)
        
        frame = self.flip_if_needed(frame)
        # Picamera2返回RGB格式，转换为cv2需要的BGR格式
        frame_bgr = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
        ret, jpeg = cv2.imencode('.jpg', frame_bgr)
        return jpeg.tobytes()

3. 关键适配点说明 #

• 格式转换：Picamera2默认捕获RGB格式帧，而OpenCV的imencode需要BGR格式，因此添加cv2.COLOR_RGB2BGR转换。
• 线程化读取：树莓派Zero性能有限，线程化读取可避免主线程被阻塞，保证流的稳定性。
• 资源安全：使用线程锁frame_lock避免多线程下的帧数据竞争，确保读取的帧完整。
• 分辨率调整：如果流仍卡顿，可将配置中的size改为(320, 240)进一步降低负载。

4. 测试验证 #

添加测试代码验证功能是否正常：

if __name__ == "__main__":
    camera = VideoCamera(flip=True)
    try:
        while True:
            jpeg_data = camera.get_frame()
            # 将帧写入文件查看效果
            with open("camera_test.jpg", "wb") as f:
                f.write(jpeg_data)
            time.sleep(1)
    except KeyboardInterrupt:
        del camera

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