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 环境后执行:
pip install ultralytics如果下载很慢,可以使用镜像源:
pip install ultralytics -i https://pypi.tuna.tsinghua.edu.cn/simple安装完成后,可以测试一下:
yolo version如果能输出版本号,说明命令行工具可以正常使用。
3、命令行预测
最简单的预测方式是直接使用 yolo 命令:
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)运行:
python predict.py参数说明:
source:输入图片路径,也可以是文件夹或视频。save=True:保存预测结果。conf=0.25:置信度阈值,低于这个值的检测框不显示。
5、查看检测结果
如果运行成功,终端里会显示结果保存位置。一般是:
runs/detect/predict打开里面的图片,如果能看到检测框和类别名称,说明第一次预测成功。
6、常见问题
(1)找不到模型文件
如果报错提示找不到 yolo11n.pt,先检查当前命令行所在目录:
pwdWindows 可以使用:
cd最简单的解决方式是使用绝对路径。
(2)CUDA 不可用
可以在 Python 中检查:
import torchprint(torch.cuda.is_available())如果输出 False,说明当前 PyTorch 没有识别到 GPU。这时先不要急着改 YOLO,应该回头检查显卡驱动、CUDA 版本和 PyTorch 版本是否匹配。
(3)预测速度很慢
如果使用 CPU 推理,速度慢是正常的。可以先用小模型 yolo11n.pt 测试,等流程跑通后再考虑更大的模型。
总结
YOLO11 学习建议按这个顺序来:
环境配置 -> 预训练模型预测 -> 理解输出结果 -> 准备自己的数据集 -> 训练模型今天先完成第一步预测,只要能跑通 test.jpg,后面训练自己的数据集就更容易排查问题。
文章分享
如果这篇文章对你有帮助,欢迎分享给更多人!













