加载已训练好的目标检测YOLOv8,v5,v3,v6模型，对数据集中某张图片中的object打上方框、标出类别，并将图片保存到本地

最新推荐文章于 2024-07-15 23:27:23 发布

德彪稳坐倒骑驴

最新推荐文章于 2024-07-15 23:27:23 发布

阅读量1.6k

点赞数

文章标签：目标检测 YOLO 人工智能

本文链接：https://blog.csdn.net/albert233333/article/details/131951431

版权

该教程介绍了如何在Python环境中运用Ultralytics的YOLOv8模型对图像进行标注。首先，从Ultralytics导入YOLO库，然后加载预训练模型yolov8n.pt。接着，打开一张名为bus.jpg的照片进行标注，并将标注后的图像保存至本地。最后，打印出结果。

摘要由CSDN通过智能技术生成

参考的教程：Python - Ultralytics YOLOv8 Docs

在与ultralytics代码同一层级下新建 predict.py

里面写下面的内容。运行即可

from ultralytics import YOLO
from PIL import Image
import cv2

# 加载计划使用的模型
model = YOLO("yolov8n.pt")  # load a pretrained model (recommended for training)

# 加载被标注的照片
im1 = Image.open("bus.jpg")

# 完成标注并保存到本地
results = model.predict(source=[im1], save=True)  # save plotted images
# results = model.predict(source=[im1], save=True)  # save plotted images

print(results)