在Google Colab GPU环境运行cv2摄像头代码致会话崩溃求助

核心崩溃原因 #

• 代码拼写错误：最后一行的sorce.release()应为source.release()，未定义的变量会引发异常导致崩溃。
• Colab环境限制：Colab是云端无头（无桌面）环境，cv.namedWindow和cv.waitKey依赖本地图形界面，无法在云端正常运行，触发崩溃。
• 摄像头访问方式错误：cv.VideoCapture(0)会尝试访问Colab云端机器的摄像头，而云端机器无外接摄像头设备，读取失败引发后续异常。

修正后的可行方案 #

要在Colab中访问本地摄像头，需借助Colab的浏览器摄像头授权API，同时用google.colab.patches.cv2_imshow替代cv.imshow显示图像（云端无图形界面，后者无法工作）。

完整可运行代码 #

import cv2 as cv
import numpy as np
from google.colab.patches import cv2_imshow
from IPython.display import display, Javascript
from google.colab.output import eval_js
from base64 import b64decode
import time

# 定义获取摄像头单帧画面的函数
def get_frame():
    js = Javascript('''
        async function getFrame() {
            const stream = await navigator.mediaDevices.getUserMedia({video: true});
            const video = document.createElement('video');
            video.srcObject = stream;
            await video.play();
            const canvas = document.createElement('canvas');
            canvas.width = video.videoWidth;
            canvas.height = video.videoHeight;
            const ctx = canvas.getContext('2d');
            ctx.drawImage(video, 0, 0);
            stream.getVideoTracks()[0].stop();
            return canvas.toDataURL('image/jpeg', 0.8);
        }
    ''')
    display(js)
    data = eval_js('getFrame()')
    binary = b64decode(data.split(',')[1])
    return cv.imdecode(np.frombuffer(binary, np.uint8), cv.IMREAD_COLOR)

# 实时显示摄像头画面（运行10秒后自动停止）
start_time = time.time()
while time.time() - start_time < 10:
    frame = get_frame()
    cv2_imshow(frame)

# 注：Colab无法通过cv.waitKey监听键盘输入，可修改时间限制或手动点击运行框的停止按钮终止程序

关键说明 #

• 运行代码时会弹出摄像头授权请求，允许浏览器访问本地摄像头即可。
• 必须使用cv2_imshow而非cv.imshow，后者在云端无头环境下无法正常渲染图像。

内容的提问来源于stack exchange，提问作者Mr. Hosseini