如何使用摄像头作为pygame的屏幕？

import pygame
import cv2
import numpy as np

pygame.init()

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

camera = cv2.VideoCapture(0)

while True:
    # 读取摄像头画面
    ret, frame = camera.read()
    
    # 将OpenCV图像转换为Pygame图像
    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
    frame = np.rot90(frame)
    frame = pygame.surfarray.make_surface(frame)
    
    # 显示摄像头画面
    screen.blit(frame, (0, 0))
    pygame.display.flip()
    
    # 检测退出事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            camera.release()
            pygame.quit()
            exit(0)