加载已训练好的目标检测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)
相关推荐
JxWang0523 分钟前
pandas计算某列每行带有分隔符的数据中包含特定值的次数
人工智能
能源系统预测和优化研究23 分钟前
创新点解读:基于非线性二次分解的Ridge-RF-XGBoost时间序列预测(附代码实现)
人工智能·深度学习·算法
执笔论英雄28 分钟前
【RL】ROLL下载模型流程
人工智能·算法·机器学习
لا معنى له31 分钟前
目标分割介绍及最新模型----学习笔记
人工智能·笔记·深度学习·学习·机器学习·计算机视觉
carver w40 分钟前
one-hot编码
人工智能
邮一朵向日葵1 小时前
企查查开放平台MCP:为AI智能体注入精准商业数据,驱动智能决策新时代
大数据·人工智能
沃达德软件1 小时前
智能警务视频侦查系统
大数据·人工智能·数据挖掘·数据分析·实时音视频·视频编解码
说私域1 小时前
链动2+1模式AI智能名片S2B2C商城小程序中电商直播的应用机制与价值创新研究
人工智能·小程序
北邮刘老师1 小时前
【智能体互联协议解析】身份码-智能体的身份证号
网络·人工智能·大模型·智能体·智能体互联网
Wulida0099912 小时前
【目标检测】基于改进YOLOv13-C3k2-DWR的铲斗定位系统研究
人工智能·yolo·目标检测