使用 YOLO 进行目标检测：如何提取人物图像

关注

最新推荐文章于 2026-08-10 07:57:08 发布 · 1.4k 阅读

原创 ·

本内容遵循 CC 4.0 BY-SA 版权协议

利用yolov5进行目标检测，并将检测到的目标裁剪出来写在前面：关于yolov5的调试运行在这里不做过多赘述，有关yolov5的调试运行请看：本文章主要讲解的是裁剪。需求：识别图片中的人物并将其裁剪出来如果只需识别人物的话，那么只需在yolov5中设定参数即可，例如使用命令行运行时：即为将参数设置为只识别人。此外需要将检测到的目标裁剪出来还需要目标的中心点坐标以及目标的宽高，因此需要保存目标的信息。实现人物检测并保存检测到的目标信息，运行命令为：在保存目标信息的tx文件中，例如下图：0 0.682078 0.495935 0.613014 0.99187 阅读详情

YOLO（You Only Look Once）是一种流行的用于对象检测的开源神经网络模型。在这篇文章中，我们将解释如何使用 YOLO 提取一堆人（或至少一个人）的图像。

首先，我们需要安装 YOLO 库和依赖项。为此，我们将使用 pip 包管理器并安装以下库：

pip install numpy
pip install opencv-python
pip install tensorflow
pip install keras

接下来我们从官网下载预训练好的YOLO权重和配置文件。这些文件可以在https://pjreddie.com/darknet/yolo/找到。

一旦我们有了权重和配置文件，我们就可以使用它们对我们的图像执行对象检测。

以下是如何使用 YOLO 检测图像中人物的示例：

import cv2
import numpy as np

# Load YOLO model
net = cv2.dnn.readNet("./yolov3.weights", "./darknet/cfg/yolov3.cfg")

# Define input image
image = cv2.imread("image.jpg")

# Get image dimensions
(height, width) = image.shape[:2]

# Define the neural network input
blob = cv2.dnn.blobFromImage(image, 1 / 255.0, (416, 416), swapRB=True, crop=False)
net.setInput(blob)

# Perform forward propagation
output_layer_name = net.getUnconnectedOutLayersNames()
output_layers = net.forward(output_layer_name)

# Initialize list of detected people
people = []

# Loop over the output layers
for output in output_layers:
    # Loop over the detections
    for detection in output:
        # Extract the class ID and confidence of the current detection
        scores = detection[5:]
        class_id = np.argmax(scores)
        confidence = scores[class_id]

        # Only keep detections with a high confidence
        if class_id == 0 and confidence > 0.5:
            # Object detected
            center_x = int(detection[0] * width)
            center_y = int(detection[1] * height)
            w = int(detection[2] * width)
            h = int(detection[3] * height)

            # Rectangle coordinates
            x = int(center_x - w / 2)
            y = int(center_y - h / 2)

            # Add the detection to the list of people
            people.append((x, y, w, h))

# Draw bounding boxes around the people
for (x, y, w, h) in people:
    cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2)

# Show the image
cv2.imshow("Image", image)
cv2.waitKey(0)

在这个例子中，我们首先使用cv2.dnn.readNet方法加载 YOLO 模型。然后，我们定义输入图像并获取其尺寸。

接下来，我们通过从图像创建一个斑点并对其像素值进行归一化来定义神经网络输入。

最后，我们执行前向传播并在输出层上循环，以提取检测并提取人周围的边界框坐标（我们的兴趣在于人检测）。

然后我们可以使用cv2.rectangle方法在原始图像上的人周围绘制边界框。

运行此代码后，你应该能够看到检测到的任何人周围都带有边界框。

YOLO 对象检测 | 来源(https://unsplash.com/photos/kY8m5uDIW7Y)

如果你想去除重复的矩形，你可以使用 NMS，例如：

cv2.dnn.NMSBoxes(boxes, confidences, score_threshold=0.5, nms_threshold=0.4)

此外，可以为每个检测到的矩形添加一个标题：

cv2.putText(image, label, (x, y + 30), font, 2, color, 3)

输出是：

YOLO 对象检测（带标签的对象）| 来源(https://unsplash.com/photos/kY8m5uDIW7Y)

现在你可以轻松地计算图像中的人数。

☆ END ☆

如果看到这里，说明你喜欢这篇文章，请转发、点赞。微信搜索「uncle_pn」，欢迎添加小编微信「 woshicver」，每日朋友圈更新一篇高质量博文。

↓扫描二维码添加小编↓

标签

#YOLO #目标检测 #计算机视觉 #深度学习 #python

用yolov5图像分割做人物抠像用yolov5的6.2版本新推出的图像分割模型来做人物抠像，速度很快效果也不错！阅读详情

使用 YOLO 进行目标检测：如何提取人物图像

相关推荐