yolov5游客计数

### 使用YOLOv5进行游客人数统计

YOLOv5 是一种高效的实时目标检测算法，在处理视频流中的对象识别方面表现出色。为了实现基于 YOLOv5 的游客计数功能，可以按照以下方式构建解决方案[^1]。

#### 数据集准备
首先需要收集并标注用于训练的数据集。对于人员计数任务来说，通常会使用 COCO 或者自定义数据集中的人类标签来训练模型。如果特定场景下有特殊需求，则建议创建专门针对该环境下的定制化数据集以提高准确性。

#### 模型微调
由于预训练权重已经在大量图像上进行了优化，因此可以直接加载官方提供的预训练版本，并在此基础上做少量调整即可适应新的应用场景。具体操作如下：

- 下载 YOLOv5 官方仓库；
- 修改配置文件 `yolov5s.yaml` 中类别数量为所需数目（一般情况下保持默认值不变）；
- 将个人采集到的新样本加入原有数据集中一起参与迭代更新过程；

通过这种方式能够快速获得适用于新领域内物体检测效果良好的神经网络结构实例。

#### 游客计数逻辑设计
当完成上述准备工作之后就可以着手编写实际业务逻辑部分了。这里给出一段简单的 Python 代码片段作为参考，它展示了如何利用 OpenCV 和 PyTorch 来读取摄像头画面并对其中的人物进行追踪与计数:

import torch
from pathlib import Path
import cv2


def main():
    weights_path = 'path/to/best.pt'  # 替换成自己训练好的模型路径
    source_video = "0"  # 表示打开本地摄像头, 若需指定其他视频源则替换为此处参数
    
    model = torch.hub.load('ultralytics/yolov5', 'custom', path=weights_path)

    cap = cv2.VideoCapture(int(source_video))

    while True:
        ret, frame = cap.read()
        
        results = model(frame)
        detections = results.pandas().xyxy[0]

        person_count = (detections['name'] == 'person').sum()

        annotated_frame = results.render()[0]
        text = f"People Count: {person_count}"
        position = (10, 30)
        font_scale = 1
        color = (255, 0, 0)  
        thickness = 2
        
        cv2.putText(annotated_frame, text=text, org=position,
                    fontFace=cv2.FONT_HERSHEY_SIMPLEX,
                    fontScale=font_scale, color=color, thickness=thickness)

        cv2.imshow("Frame", annotated_frame)

        key = cv2.waitKey(1) & 0xFF
        if key == ord("q"):
            break
            
main()

这段程序实现了从摄像头获取每一帧的画面信息并通过已训练完毕的目标检测器对其进行分析处理的功能。每当发现有人出现在视野范围内时就会增加对应位置上的计数值，并最终显示于屏幕之上供观察者查看。