• 智能AI问答!更聪明、更完整、原创作!
  • 一款基于GPT的超级AI机器人,可以回答问题、聊天、创作、写文案!

可以yolov8分类图像识别代码吗

发布于 2023-10-29 20:49:17

可以的,以下是使用YOLOv8进行图像识别的代码示例:

import cv2
import torch

# 加载YOLOv8模型
model = torch.hub.load('ultralytics/yolov8', 'yolov8')

# 读取图像
image_path = 'path/to/your/image.jpg'
image = cv2.imread(image_path)

# 将图像转换为YOLOv8所需的格式
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image = image / 255.0
image = torch.tensor(image).permute(2, 0, 1).float().unsqueeze(0)

# 进行目标检测
results = model(image)

# 解析检测结果
labels = results.xyxy[0][:, 6].tolist()
confidences = results.xyxy[0][:, 4].tolist()
boxes = results.xyxy[0][:, :4].tolist()

# 在图像上绘制检测结果
for label, confidence, box in zip(labels, confidences, boxes):
    label = int(label)
    confidence = float(confidence)
    box = [int(x) for x in box]
    x1, y1, x2, y2 = box
    cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
    cv2.putText(image, f'{label}: {confidence:.2f}', (x1, y1 - 5), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)

# 显示结果
cv2.imshow('YOLOv8 Object Detection', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

请确保已经安装了torchopencv-python库。如果没有安装,可以使用以下命令进行安装:

pip install torch opencv-python
0 条评论

发布
问题

在线
客服