加载已训练好的目标检测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)
相关推荐
云烟成雨TD几秒前
Spring AI Alibaba 1.x 系列【54】Interrupts 中断机制:析动态中断源码分析
java·人工智能·spring
布吉岛的石头1 分钟前
Java 程序员第 29 阶段-01:大模型微调入门:小样本业务适配方案
java·开发语言·人工智能
什么半岛铁盒2 分钟前
LangChain 入门与架构:快速搭建你的第一个 AI 应用
人工智能·架构·langchain
松☆2 分钟前
torchair:昇腾PyTorch适配层生态协作深度解读
人工智能·pytorch·python
科技那些事儿5 分钟前
一眸科技 | 情感认知智能,让AI更懂人心
人工智能·科技
java1234_小锋6 分钟前
Spring AI 2.0 开发Java Agent智能体 - 多模态支持
java·人工智能·spring
无心水6 分钟前
【Harness:全局认知】3、Harness 如何改写软件交付规则?从 52.8% 到 66.5% 的跨越背后
人工智能·性能优化·openclaw·养龙虾·harness·hermes·honcho
放下华子我只抽RuiKe56 分钟前
React 从入门到生产(五):状态管理选型
前端·javascript·人工智能·深度学习·react.js·前端框架·ecmascript
前端若水7 分钟前
使用 IndexedDB 在客户端存储对话记录
java·前端·人工智能·python·机器学习
未来智慧谷9 分钟前
汉中首家OPC社区正式成立!未来智慧谷联合京东科技(汉中)数字经济产业园打造“一人公司”企业新生态
大数据·人工智能·科技