如何在PyGame画布上从PyRealSense2绘制视频帧？

import pygame
import pyrealsense2 as rs

pygame.init()
width, height = 640, 480
screen = pygame.display.set_mode((width, height))
canvas = pygame.Surface((width, height))

pipeline = rs.pipeline()
config = rs.config()
config.enable_stream(rs.stream.depth, width, height, rs.format.z16, 30)
config.enable_stream(rs.stream.color, width, height, rs.format.rgb8, 30)
pipeline.start(config)

while True:
    # 获取RealSense相机的帧数据
    frames = pipeline.wait_for_frames()
    depth_frame = frames.get_depth_frame()
    color_frame = frames.get_color_frame()

    # 将颜色帧数据转换为PyGame可用的格式
    color_image = pygame.image.frombuffer(color_frame.get_data(), (width, height), 'RGB')

    # 绘制视频帧到画布上
    canvas.blit(color_image, (0, 0))

    # 将画布绘制到PyGame窗口上
    screen.blit(canvas, (0, 0))
    pygame.display.flip()

    # 处理事件，例如退出事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            pipeline.stop()
            exit(0)