当前位置: 首页 > news >正文

YOLOv12 项目部署指南! 含报错解决

前段时间,yolo12横空出世,所以去官网看了下官方视频和相关论文。并成功在本地复现项目

目录

1. 环境准备

2. 项目下载

3. 安装依赖

4. 下载预训练模型

5. 运行演示应用

6. 使用说明

7. 导出当前环境依赖

常见问题解决

验证安装


1. 环境准备

首先创建并激活虚拟环境:

conda create -n yolov12 python=3.11
conda activate yolov12

2. 项目下载

cd e:\Dev_project\MyProjects
git clone https://github.com/sunsmarterjie/yolov12.git
cd yolov12

3. 安装依赖

使用国内镜像源安装依赖(推荐清华源):

1. 首先安装 PyTorch(基于您的 CUDA 12.8 环境):

pip install torch torchvision --index-url https://mirrors.tuna.tsinghua.edu.cn/pytorch-wheels/cu121

2. 安装其他必要依赖:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple matplotlib
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple gradio
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple ultralytics
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple thop
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple psutil

4. 下载预训练模型

从 YOLOv12 官方仓库下载预训练模型,放在项目根目录下。您可以选择以下任意一个模型:

  • yolov12n.pt(最小模型)
  • yolov12s.pt(小型模型)
  • yolov12m.pt(中型模型)
  • yolov12l.pt(大型模型)
  • yolov12x.pt(超大模型)

5. 运行演示应用

python app.py

运行后,浏览器会自动打开 http://127.0.0.1:7860 地址,显示 Web 界面。

6. 使用说明

  1. 在 Web 界面中:

    • 选择输入类型(图片或视频)
    • 选择要使用的模型
    • 调整图像大小和置信度阈值
    • 上传图片或视频
    • 点击 "Detect Objects" 按钮开始检测
  2. 示例图片位于 ultralytics/assets/ 目录下

7. 导出当前环境依赖

导出环境依赖到 requirements.txt:

pip freeze > requirements.txt

方便下次在其他环境部署运行。 

常见问题解决

  1. 如果遇到 CUDA 相关错误:

    • 确保已安装 NVIDIA 驱动
    • 确保 CUDA 版本与 PyTorch 版本匹配
  2. 如果遇到模块导入错误:

    • 检查是否已安装所有必要依赖
    • 尝试重新安装相应模块
  3. 如果模型下载速度慢:

    • 可以使用国内镜像源
    • 或者手动下载模型文件
  4. 如果运行时出现内存错误:

    • 尝试使用较小的模型(如 yolov12n.pt)
    • 减小输入图像尺寸

验证安装

安装完成后,可以运行以下代码验证环境:

import torch
print(f"PyTorch version: {torch.__version__}")
print(f"CUDA available: {torch.cuda.is_available()}")
print(f"CUDA version: {torch.version.cuda}")

ultralytics操作文档:YOLO12 -Ultralytics YOLO 文档

Demo体验网站:YOLOv12 Demo - a Hugging Face Space by sunsmarterjieleaf

官方训练视频:YOLO12: Train for Real-Time Object Detection - YouTube
操作文档参考:How to Train a YOLOv12 Object Detection Model on a Custom Dataset

github 项目链接:sunsmarterjie/yolov12: YOLOv12: Attention-Centric Real-Time Object Detectors

相关文章:

  • Flutter底层实现
  • Go学习笔记:基础语法3
  • 【由技及道】镜像星门开启:Harbor镜像推送的量子跃迁艺术【人工智障AI2077的开发日志010】
  • CSS+Html面试题(二)
  • python网络爬虫开发实战之爬虫基础
  • Unity自定义渲染管线(Scriptable Render Pipeline)架构设计与实现指南
  • netty中Future和ChannelHandler
  • Best practice-生产环境中加锁的最佳实践
  • Anaconda 部署 DeepSeek
  • Java 大视界 -- Java 大数据在智能政务公共服务资源优化配置中的应用(118)
  • Linux | Vim 鼠标不能右键粘贴、跨系统复制粘贴
  • 深入解析“Elaborate”——从详细阐述到精心制作的多重含义
  • 绝美焦糖暖色调复古风景画面Lr调色教程,手机滤镜PS+Lightroom预设下载!
  • LLM-初识AI
  • 自律linux 第 35 天
  • 【C++】数据结构 双链表的实现(企业存储用户数据的实现)
  • Windows逆向工程入门之MASM 数据寻址
  • GTID模块初始化简介和参数binlog_gtid_simple_recovery
  • C#数据类型及相互转换
  • GitHub获取token
  • 上海浦江游览南拓新航线首航,途经前滩、世博文化公园等景点
  • 总书记回信二周年之际,上海如何将垃圾分类深度融入城市发展?
  • 一季度支持科技创新和制造业发展减税降费及退税4241亿元
  • 19世纪的“表征之场”:弗洛伊德的精神分析椅
  • 消费维权周报丨上周涉汽车类投诉较多,涉加油“跳枪”等问题
  • 贯彻落实《生态环境保护督察工作条例》,充分发挥生态环境保护督察利剑作用