YOLO11 第一次预测图片

687 字
3 分钟
YOLO11 第一次预测图片

YOLO11 第一次预测图片#

前面已经记录过 YOLO11 的环境配置。环境装好以后,下一步最应该做的不是马上训练自己的数据集,而是先跑通一张图片的预测流程。

只要预测流程能跑通,就说明 Python 环境、依赖库、模型文件、显卡或 CPU 推理基本没有大问题。

1、准备文件目录#

可以先建立一个简单目录:

yolo_test/
├── yolo11n.pt
├── test.jpg
└── predict.py

其中:

  • yolo11n.pt 是预训练模型。
  • test.jpg 是需要检测的图片。
  • predict.py 是推理代码。

建议图片路径和模型路径先放在同一个目录中,这样可以减少路径错误。

2、安装 ultralytics#

进入对应的 conda 环境后执行:

Terminal window
pip install ultralytics

如果下载很慢,可以使用镜像源:

Terminal window
pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple

安装完成后,可以测试一下:

Terminal window
yolo version

如果能输出版本号,说明命令行工具可以正常使用。

3、命令行预测#

最简单的预测方式是直接使用 yolo 命令:

Terminal window
yolo predict model=yolo11n.pt source=test.jpg

运行完成后,一般会生成类似目录:

runs/detect/predict/

里面会有标注好检测框的图片。

4、Python 代码预测#

也可以写成 Python 脚本:

from ultralytics import YOLO
# 加载模型
model = YOLO("yolo11n.pt")
# 执行预测
results = model.predict(
source="test.jpg",
save=True,
conf=0.25
)
print("预测完成")
print(results)

运行:

Terminal window
python predict.py

参数说明:

  • source:输入图片路径,也可以是文件夹或视频。
  • save=True:保存预测结果。
  • conf=0.25:置信度阈值,低于这个值的检测框不显示。

5、查看检测结果#

如果运行成功,终端里会显示结果保存位置。一般是:

runs/detect/predict

打开里面的图片,如果能看到检测框和类别名称,说明第一次预测成功。

6、常见问题#

(1)找不到模型文件#

如果报错提示找不到 yolo11n.pt,先检查当前命令行所在目录:

Terminal window
pwd

Windows 可以使用:

Terminal window
cd

最简单的解决方式是使用绝对路径。

(2)CUDA 不可用#

可以在 Python 中检查:

import torch
print(torch.cuda.is_available())

如果输出 False,说明当前 PyTorch 没有识别到 GPU。这时先不要急着改 YOLO,应该回头检查显卡驱动、CUDA 版本和 PyTorch 版本是否匹配。

(3)预测速度很慢#

如果使用 CPU 推理,速度慢是正常的。可以先用小模型 yolo11n.pt 测试,等流程跑通后再考虑更大的模型。

总结#

YOLO11 学习建议按这个顺序来:

环境配置 -> 预训练模型预测 -> 理解输出结果 -> 准备自己的数据集 -> 训练模型

今天先完成第一步预测,只要能跑通 test.jpg,后面训练自己的数据集就更容易排查问题。

文章分享

如果这篇文章对你有帮助,欢迎分享给更多人!

YOLO11 第一次预测图片
https://blog.tjcheng.cn/posts/daily/2026-06-29/yolo11-predict-first/
作者
TengJinCheng
发布于
2026-06-29
许可协议
CC BY-NC-SA 4.0
Profile Image of the Author
TengJinCheng
Hello, I'm Teng.
公告
欢迎来到TengJinCheng工作铺!
音乐
封面

音乐

暂未播放

0:000:00
暂无歌词
分类
标签
站点统计
文章
157
分类
7
标签
114
总字数
109,939
运行时长
0 天
最后活动
0 天前
站点信息
构建平台
Local
博客版本
Firefly v6.13.4
文章许可
CC BY-NC-SA 4.0

文章目录