可以使用 OpenCV 库来调用 USB 摄像头并获取实时视频帧。然后,将视频帧转换为模型需要的输入格式,然后使用 onnxruntime 进行推理。最后,将预测的标签在窗口中显示出来即可。
下面是一个示例代码:
import cv2
import onnxruntime as rt
import numpy as np
# 调用摄像头并获取视频帧
cap = cv2.VideoCapture(0)
# 加载模型
sess = rt.InferenceSession("model.onnx")
# 循环显示每一帧
while True:
# 获取视频帧