写一段python程序，用模型（已有）对摄像头进行推理，根据模型的识别框，计算大小，距离，类别

### 实现方案为了完成基于已有深度学习模型对摄像头捕获图像进行实时推理的任务，可以通过 Python 编写程序并利用 OpenCV 和 PyTorch 来加载预训练的 YOLOv5 模型。以下是具体实现方法： #### 加载模型首先需要导入必要的库，并加载已有的 YOLOv5 模型。这一步通常通过 `torch.hub` 或本地路径加载 `.pt` 文件来完成。 ```python import cv2 import torch from PIL import Image, ImageDraw # 加载YOLOv5模型 model = torch.hub.load('ultralytics/yolov5', 'custom', path='path/to/best.pt') # 替换为实际模型路径[^1] ``` #### 摄像头捕获与推理使用 OpenCV 的 `cv2.VideoCapture(0)` 方法打开默认摄像头，并逐帧读取视频流。每帧图像传递给模型进行推理，返回检测结果。 ```python cap = cv2.VideoCapture(0) while cap.isOpened(): ret, frame = cap.read() if not ret: break # 将帧转换为PIL格式供模型处理 img = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)) # 进行推理 results = model(img) # 推理过程[^1] # 获取检测结果 detections = results.pandas().xyxy[0].to_dict(orient="records") # 结果转为字典列表[^1] # 绘制边界框和标注信息 draw = ImageDraw.Draw(img) for detection in detections: bbox = [ detection['xmin'], detection['ymin'], detection['xmax'], detection['ymax'] ] class_name = detection['name'] confidence = round(detection['confidence'], 2) # 计算检测框宽度和高度 box_width = int(bbox[2] - bbox[0]) box_height = int(bbox[3] - bbox[1]) # 假设距离估算逻辑（简单线性关系） focal_length = 800 # 镜头焦距假设值 (单位：像素)[^3] object_width_in_pixels = box_width real_object_width = 20 # 物体真实宽度 (单位：厘米)[^3] distance = (real_object_width * focal_length) / object_width_in_pixels label = f"{class_name} {confidence} Dist:{distance:.2f}cm W:{box_width}px H:{box_height}px" draw.rectangle(bbox, outline=(0, 255, 0), width=2) draw.text((bbox[0], bbox[1]), label, fill=(255, 255, 255)) # 显示结果 result_frame = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) cv2.imshow("Detection", result_frame) if cv2.waitKey(1) & 0xFF == ord('q'): break cap.release() cv2.destroyAllWindows() ``` #### 关键点解析 - **模型加载**：通过 `torch.hub.load()` 函数加载自定义训练的 YOLOv5 模型[^1]。 - **摄像头捕获**：使用 OpenCV 提供的 `VideoCapture` 功能获取实时画面[^3]。 - **推理与结果提取**：调用模型对象直接传入图像即可获得预测结果；结果可通过 Pandas DataFrame 方便地转化为可操作的数据结构[^1]。 - **边界框绘制**：借助 Pillow 库中的绘图功能，在原图上标记检测框及其属性。 - **距离计算**：采用简单的三角测距法估计物体距离相机的距离。 --- ###

创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

下一篇 RS232 python代码，将天平和电脑连接，电脑每30分钟记录一次天平的示数，生成文本

目录

写一段python程序，用模型（已有）对摄像头进行推理，根据模型的识别框，计算大小，距离，类别

Python内容推荐

基于Python+Open CV的手势识别算法设计

Python+Dlib库实现人脸检测、人脸识别、人脸数量检测和人脸距离检测.zip

视频+摄像头 使用 Python+OpenCV+OpenPose 实现人体形态算法识别

基于python的人脸识别

基于python的行人重识别代码

基于OpenCV-Python版本的人脸模型训练与识别

python的pyqt5和vc2写的调取电脑摄像头程序.

Python-Opencv下摄像头颜色识别

python+openCV调用摄像头拍摄和处理图片的实现

Python3利用Dlib19.7实现摄像头人脸识别的方法

python手势识别上下左右(SVM+HOG特征)

pycharm工程python调用OpenCV实现USB摄像头实时人脸检测

python语言使用opencv实现摄像头人脸采集、学习、识别等功能

python 摄像头二维码识别

python 用opencv调用训练好的模型进行识别的方法

Python 颜色识别

python实现简单颜色识别程序

Python写的捕获摄像头程序。

python+opencv实现动态物体识别

Python+OpenCV采集本地摄像头的视频

【Web服务器运维】基于Nginx与Keepalived的高可用负载均衡系统部署：实现故障自动切换与虚拟IP管理

三电平Buck直流变换器双闭环、单闭环、开环仿真【附参考文献】（Simulink仿真实现）

2026年AI大模型面试题整理（含答案）.zip

国央企创新负责人如何实现科技成果的全流程数字化管理？.docx

政府科技管理者如何保障科技成果评价的公正透明？.docx

MATLAB/Simulink直流电机三闭环仿真与PI及PWM控制

【MQTT协议20个致命误区】：20年IoT架构师亲历的17次生产事故复盘与9条黄金避坑法则（仅限内部技术圈流传）

切换Dify的嵌入模型后，旧文件还能直接用吗？

石墨烯超表面三次谐波增强COMSOL仿真技术及应用研究

Python算法环境三系战争终局方案：venv_poetry_conda冲突溯源图谱（基于37个真实pipdeptree快照）+ 隔离策略决策树（准确率98.6%）

视频+摄像头使用 Python+OpenCV+OpenPose 实现人体形态算法识别