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

参考的教程:Python - Ultralytics YOLOv8 Docs

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

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

python 复制代码
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)
相关推荐
badfl14 分钟前
AI漫剧技术方案拆解:NanoBanana+Sora视频生成全流程
人工智能·ai·ai作画
杭州杭州杭州28 分钟前
李沐动手学深度学习笔记(4)---物体检测基础
人工智能·笔记·深度学习
小二·1 小时前
Python Web 开发进阶实战(终章):从单体应用到 AI 原生生态 —— 45 篇技术演进全景与未来开发者生存指南
前端·人工智能·python
极客小云1 小时前
【基于 PyQt6 的红外与可见光图像配准工具开发实战】
c语言·python·yolo·目标检测
秋名山大前端1 小时前
AI数字孪生本体智能技术方案
人工智能·aigc·数据可视化
集和诚JHCTECH1 小时前
边缘智能,触手可及|BRAV-7821高能效AI边缘计算系统正式发布
大数据·人工智能·边缘计算
新缸中之脑2 小时前
现代开发者的工具箱 (2026)
人工智能
才兄说2 小时前
机器人租售出场准?会卡节点上
人工智能·机器人
救救孩子把2 小时前
64-机器学习与大模型开发数学教程-5-11 本章总结与习题
人工智能·机器学习
救救孩子把2 小时前
55-机器学习与大模型开发数学教程-5-2 梯度下降法(GD)与随机梯度下降(SGD)
人工智能·机器学习