如何在Kaggle中调用摄像头？本地可用代码在Kaggle报错

你遇到的核心问题是Kaggle的云端运行环境没有物理摄像头设备：本地环境能直接调用/dev/video0这类本地摄像头，但Kaggle服务器没有挂载任何摄像头硬件，所以不管修改cv2.VideoCapture()的索引为0、1还是2，都找不到可用设备。这会导致cap.read()返回的frame是空值，最终触发imwrite的断言错误。另外cv2.imshow在Kaggle的无界面云端环境下也无法正常运行，因为没有图形显示层支持。

方案1：提前采集图片并上传到Kaggle #

既然本地能正常采集图像，直接在本地运行你的代码生成图片，然后把图片文件夹上传到Kaggle的数据集或当前项目的输入目录，之后在Kaggle代码里直接读取这些图片即可，这是最直接高效的方式。

方案2：在Kaggle中通过浏览器调用本地摄像头采集 #

Kaggle支持通过浏览器Web API调用你本地的摄像头，需要用HTML/JS结合Python交互组件实现，示例代码如下：

from IPython.display import display, Javascript
from base64 import b64decode
import os
import uuid

def take_photo(filename='photo.jpg', quality=0.8):
    js = Javascript('''
        async function takePhoto(quality) {
            const div = document.createElement('div');
            const capture = document.createElement('button');
            capture.textContent = 'Capture';
            div.appendChild(capture);

            const video = document.createElement('video');
            video.style.display = 'block';
            const stream = await navigator.mediaDevices.getUserMedia({video: true});

            document.body.appendChild(div);
            div.appendChild(video);
            video.srcObject = stream;
            await video.play();

            // 等待用户点击拍照按钮
            await new Promise((resolve) => capture.onclick = resolve);

            const canvas = document.createElement('canvas');
            canvas.width = video.videoWidth;
            canvas.height = video.videoHeight;
            canvas.getContext('2d').drawImage(video, 0, 0);
            stream.getVideoTracks()[0].stop();
            div.remove();
            return canvas.toDataURL('image/jpeg', quality);
        }
    ''')
    display(js)
    data = eval_js('takePhoto({})'.format(quality))
    binary = b64decode(data.split(',')[1])
    with open(filename, 'wb') as f:
        f.write(binary)
    return filename

# 设置图片保存路径
IMAGES_PATH = './collected_images'
os.makedirs(IMAGES_PATH, exist_ok=True)
number_images = 5

# 批量采集图片
for imgnum in range(number_images):
    print(f'Collecting image {imgnum}')
    imgname = os.path.join(IMAGES_PATH, f'{str(uuid.uuid1())}.jpg')
    take_photo(imgname)
    print(f'Saved to {imgname}')

这段代码会在Kaggle Notebook中显示一个拍照按钮，点击后调用你本地的摄像头拍摄照片，并将照片保存到Kaggle的运行环境中。

原代码的适配修正（针对云端环境） #

如果要保留原代码的逻辑框架，需要加入空帧判断，并替换无效的cv2.imshow：

import cv2
import os
import uuid
import time
from IPython.display import Image, display

cap = cv2.VideoCapture(0)

# 先检查摄像头是否成功打开
if not cap.isOpened():
    print("无法打开摄像头设备")
    exit()

for imgnum in range(number_images):
    print('Collecting image {}'.format(imgnum))
    ret,frame = cap.read()
    # 检查是否成功读取帧
    if not ret or frame is None:
        print("无法读取摄像头帧，终止采集")
        break
    imgname = os.path.join(IMAGES_PATH,f'{str(uuid.uuid1())}.jpg')
    cv2.imwrite(imgname,frame)
    # Kaggle中无法使用cv2.imshow，改用IPython组件显示图片
    display(Image(imgname))
    time.sleep(1)
    
    # Kaggle中cv2.waitKey无效，可注释或替换为其他终止逻辑
    # if cv2.waitKey(1) & 0xFF == ord('q'):
    #     break
cap.release()
cv2.destroyAllWindows()

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