帮我写一份yolov8的ai识别代码
时间: 2024-12-21 17:22:13 浏览: 3
YOLOv8 (You Only Look Once version 8) 是一种先进的目标检测算法,基于YOLO系列,它结合了高效的实时性能和较高的精度。以下是简化的YOLOv8人工智能图像识别代码片段,使用Python和Darknet库:
```python
import cv2
from yolov8 import YOLO
# 加载模型
yolo = YOLO()
# 读取图片或视频帧
image_path = "path_to_your_image.jpg"
img = cv2.imread(image_path)
# 进行目标检测
outputs = yolo.detect(img)
# 打印和绘制框
for output in outputs:
class_id, confidence, bbox = output
if confidence > threshold: # 设置阈值过滤弱预测
x, y, w, h = bbox
cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 2)
text = f"Class {class_id}: {confidence*100:.2f}%"
cv2.putText(img, text, (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
# 显示结果
cv2.imshow("YOLOv8 Detection", img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
注意:
1. 你需要安装`yolov8`库,可以通过pip进行安装:`pip install yolov8`
2. `threshold`是一个自定义的阈值,用于确定是否显示检测到的对象。
3. 确保已经下载了预训练的YOLOv8权重文件,并将其路径设置在`yolo.detect()`函数中。
阅读全文