yolov5人脸识别完整代码

YOLOv5是一种基于深度学习的目标检测算法，用于实现人脸识别。以下是YOLOv5人脸识别的完整代码示例：

import torch
from PIL import Image

# 加载YOLOv5模型
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')

# 加载图像
image = Image.open('image.jpg')

# 进行人脸识别
results = model(image)

# 打印识别结果
results.print()

# 可视化识别结果
results.show()

在上述代码中，首先使用`torch.hub.load`函数加载了YOLOv5模型。然后，使用`Image.open`函数加载了待识别的图像。接下来，通过调用模型的`__call__`方法对图像进行人脸识别，将结果保存在`results`变量中。最后，通过调用`print`和`show`方法分别打印和可视化识别结果。

需要注意的是，运行上述代码前需要确保已经安装了`torch`和`PIL`库，并且已经下载了YOLOv5模型的权重文件。