加载已训练好的目标检测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)
相关推荐
红豆和绿豆7 分钟前
大语言模型的发展-OPENBMB
人工智能·语言模型·自然语言处理
ITdgr16 分钟前
更高效的搜索工具,国内免费好用的AI智能搜索引擎工具
人工智能·搜索引擎
fydw_71527 分钟前
PyTorch 激活函数及非线性变换详解
人工智能·pytorch·python
Apache Flink30 分钟前
Apache Flink 流批融合技术介绍
人工智能·flink·apache
南 阳40 分钟前
阿里开源多模态大模型Ovis1.6,重塑出海电商AI格局
网络·人工智能·chatgpt
HyperAI超神经1 小时前
在线教程丨1 步生成 SOTA 级别图像,Hyper-SD 一键启动教程上线!
人工智能·机器学习·stable diffusion
学习前端的小z1 小时前
【AI视频】Runway:Gen-2 运镜详解
人工智能·aigc·音视频
魔力之心1 小时前
人工智能与机器学习原理精解【23】
人工智能·机器学习
虚假程序设计1 小时前
pythonnet python图像 C# .NET图像 互转
开发语言·人工智能·python·opencv·c#·.net
AI王也2 小时前
ChatGPT 4o 使用指南 (9月更新)
人工智能·chatgpt·prompt·aigc