加载已训练好的目标检测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)
相关推荐
机 _ 长16 分钟前
05 智算运维|8 张卡怎么才算「捆成一张」:NCCL 实测与拓扑
运维·人工智能·算法·智算运维
用户43567829363737 分钟前
五线并行认知架构:从算力内卷到思考秩序的底层革命
人工智能
一 铭43 分钟前
WrenAI 详解:给 AI Agent 的开源 GenBI 引擎
人工智能·开源
say_fall1 小时前
【AI大模型】MCP 模型上下文协议:客户端-服务器架构、JSON-RPC 消息与能力协商、工具发现与调用链路、与函数调用的分工
服务器·人工智能·架构
早起CaiCai1 小时前
【2023-01 OM】利用深度学习算法融合异质观测实现海洋场降尺度
人工智能·深度学习·算法
周杰伦的稻香1 小时前
AI 助手 2.0:多轮对话、卡片化与文章页感知
人工智能
Martina_03211 小时前
AI生成3D场景从白模换成精模后镜头全偏了?用6步检查单位、枢轴与层级替换
图像处理·人工智能·3d·ai·自然语言处理·材质·1024程序员节
狒狒AI2 小时前
手绘白板动画的生产线拆解:AI 出图、笔迹对齐、镜头渲染七道工序(附 21 个开源脚本)
人工智能·aigc
用户2763965765102 小时前
上手 GitHub 官方 Spec Kit:AI 编码别再裸奔,先写 spec 再出码,SDD规约开发
人工智能
Lucas_coding2 小时前
【Agent】Agent 知识地图:核心概念、工程方法与产品生态
人工智能