零基础入门YOLO26:官方镜像快速上手,5分钟跑通推理Demo

张开发
2026/4/18 22:11:59 15 分钟阅读

分享文章

零基础入门YOLO26:官方镜像快速上手,5分钟跑通推理Demo
零基础入门YOLO26官方镜像快速上手5分钟跑通推理Demo1. 环境准备与快速部署YOLO26作为目标检测领域的最新成果其官方镜像已经预装了所有必要的运行环境。让我们先了解一下这个镜像的基本配置核心框架PyTorch 1.10.0CUDA版本12.1支持NVIDIA GPU加速Python版本3.9.5主要依赖torchvision、torchaudio、OpenCV等常用计算机视觉库这个镜像最大的优势在于开箱即用省去了繁琐的环境配置过程。你不需要手动安装任何依赖所有东西都已经准备就绪。2. 快速上手推理Demo2.1 激活环境与准备代码启动镜像后第一步是激活预配置的yolo环境conda activate yolo接着我们需要将代码复制到工作目录cp -r /root/ultralytics-8.4.2 /root/workspace/ cd /root/workspace/ultralytics-8.4.2这个步骤确保我们的修改不会因为容器重启而丢失。2.2 运行推理示例镜像已经内置了一个简单的推理脚本我们可以直接运行它来测试环境是否正常工作。创建一个名为demo.py的文件from ultralytics import YOLO # 加载预训练模型 model YOLO(yolo26n.pt) # 对示例图片进行推理 results model.predict(ultralytics/assets/bus.jpg, saveTrue) print(推理完成结果保存在runs/detect/exp目录下)保存后运行python demo.py这个简单的脚本会使用内置的YOLO26模型对示例图片进行目标检测结果会自动保存在runs/detect/exp目录中。3. 理解推理参数让我们深入了解一下predict方法的主要参数model模型权重文件路径可以是.pt或.yaml文件source输入源可以是图片路径、视频路径或摄像头编号(0)save是否保存推理结果默认为Falseshow是否显示推理结果服务器环境下建议设为False如果你想尝试不同的输入源可以修改demo.py# 对视频文件进行推理 results model.predict(input.mp4, saveTrue) # 或者使用摄像头 results model.predict(0, showTrue) # 0表示默认摄像头4. 查看推理结果推理完成后你可以在终端看到类似这样的输出image 1/1 /root/workspace/ultralytics-8.4.2/ultralytics/assets/bus.jpg: 640x480 4 persons, 1 bus, 1 stop sign, Done. (0.045s) Speed: 2.0ms preprocess, 45.0ms inference, 1.0ms postprocess per image at shape (1, 3, 640, 480)这表示模型成功检测到了图片中的4个人、1辆公交车和1个停车标志。结果图片会保存在runs/detect/exp目录下你可以使用文件浏览器查看。5. 进阶使用建议5.1 使用不同的预训练模型镜像内置了多个预训练模型你可以尝试不同的模型# 使用更精确但更慢的模型 model YOLO(yolo26x.pt) # 或者使用专门用于姿态估计的模型 pose_model YOLO(yolo26n-pose.pt)5.2 调整推理参数你可以通过调整参数来优化推理效果results model.predict( sourceinput.jpg, conf0.5, # 置信度阈值 iou0.45, # IOU阈值 imgsz640 # 推理尺寸 )6. 总结与下一步通过这个快速入门指南你已经学会了如何激活YOLO26镜像环境如何运行简单的推理Demo如何理解和使用推理参数如何查看和解读推理结果接下来你可以尝试使用自己的图片或视频进行推理测试探索镜像内置的其他预训练模型学习如何训练自定义数据集获取更多AI镜像想探索更多AI镜像和应用场景访问 CSDN星图镜像广场提供丰富的预置镜像覆盖大模型推理、图像生成、视频生成、模型微调等多个领域支持一键部署。

更多文章