计算机视觉:YOLOv12安装环境

YOLOv12安装环境

一、工具软件准备

1、yolov12

1)下载yolov12主体部分

推荐官方地址:https://github.com/sunsmarterjie/yolov12

2)下载训练模型

地址:

https://github.com/sunsmarterjie/yolov12

3)安装命令和python使用基础脚本

在这个地址(https://github.com/sunsmarterjie/yolov12)下的yolo使用命令

安装命令:

conda create -n yolov12 python=3.11 supervision flash-attn

conda activate yolov12

git clone https://github.com/sunsmarterjie/yolov12 && cd yolov12

pip install -r requirements.txt

pip install -e .

总计:

命令行切换到"D:\python练习\yolo-v12\yolov12-main"路径下

运行(pip install -e .)注意命令用.符号结尾

另:pytorch清华镜像地址:

https://pypi.tuna.tsinghua.edu.cn/simple/torch/

对应python版本下载torch、torchaudio、orchvision

本地安装命令,注意在虚拟环境下,安装包所在路径下执行命令

(YOLOv12) D:\>pip install torch-2.2.2+cpu-cp311-cp311-win_amd64.whl

复制代码
pip torchaudio-2.2.2+cpu-cp311-cp311-win_amd64.whl
复制代码
pip torchvision-0.17.2+cpu-cp311-cp311-win_amd64.whl

3、下载anaconda3虚拟环境管理工具

下载地址:(https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

4、下载pycharm软件(https://www.jetbrains.com/pycharm/download/?section=windows)

二、搭配yolo v12环境

1、前提电脑已安装anaconda3、pycharm

2、创建虚拟环境yolov12py39

1)系统命令行CMD

2)运行conda create -n yolov12py39 python=3.9

  1. pip3 install torch torchvision

4)安装yolo库

pip install yolo -i https://pypi.tuna.tsinghua.edu.cn/simple

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

  1. 命令行切换到"D:\python练习\yolo-v12\yolov12-main"路径下

运行(pip install -e .)注意命令用.符号结尾

安装成功后,pip list如下:

Validation

复制代码
from ultralytics import YOLO
复制代码
复制代码
model = YOLO('yolov12{n/s/m/l/x}.pt')
复制代码
model.val(data='coco.yaml', save_json=True)

Training

复制代码
from ultralytics import YOLO
复制代码
复制代码
model = YOLO('yolov12n.yaml')
复制代码
复制代码
# Train the model
复制代码
results = model.train(
复制代码
  data='coco.yaml',
复制代码
  epochs=600, 
复制代码
  batch=256, 
复制代码
  imgsz=640,
复制代码
  scale=0.5,  # S:0.9; M:0.9; L:0.9; X:0.9
复制代码
  mosaic=1.0,
复制代码
  mixup=0.0,  # S:0.05; M:0.15; L:0.15; X:0.2
复制代码
  copy_paste=0.1,  # S:0.15; M:0.4; L:0.5; X:0.6
复制代码
  device="0,1,2,3",
复制代码
)
复制代码
复制代码
# Evaluate model performance on the validation set
复制代码
metrics = model.val()
复制代码
复制代码
# Perform object detection on an image
复制代码
results = model("path/to/image.jpg")
复制代码
results[0].show()

Prediction

复制代码
from ultralytics import YOLO
复制代码
复制代码
model = YOLO('yolov12{n/s/m/l/x}.pt')
复制代码
model.predict()

Export

复制代码
from ultralytics import YOLO
复制代码
复制代码
model = YOLO('yolov12{n/s/m/l/x}.pt')
复制代码
model.export(format="engine", half=True)  # or format="onnx"

Demo

复制代码
python app.py
复制代码
# Please visit http://127.0.0.1:7860

本机无nvidia显卡,安装cpu版本pytorch

复制代码
运行命令:pip3 install torch torchvision

6)用pycharm打开文件

三、yolov12检测图片

1、检测人物

复制代码
import time

from ultralytics import YOLO



t1=time.time()

yolo = YOLO("./model\\yolov12n.pt", task="detect")

result = yolo(source="./picture\\bus.png", save=True)

print(f'用时{time.time()-t1}秒')

输出:

图片:

四、本文安装数据已上传到:

https://download.csdn.net/download/m0_67097444/92885088?spm=1011.2124.3001.6210

相关推荐
宋哥转AI1 分钟前
深入理解 AI Agent · 多 Agent 编排 #03:Supervisor 与 Orchestrator——Dream-SaaS 双层编排架构拆解
java·人工智能·ai
逻辑君1 分钟前
ANNA 7.6 知识智能引擎白皮书
人工智能·机器学习
叠层归一研究院2 分钟前
基于区分本源的此刻存在唯一性本体论推演
人工智能·算法
维核科技5 分钟前
三个业务共用一台服务器,显存不够用了
大数据·服务器·人工智能
听我哔哔5 分钟前
AI 漫剧制作步骤全流程分享:从剧本到成片的个人实操记录
人工智能·ai漫剧·漫剧制作
Bode_20026 分钟前
AI工具/大模型学习
人工智能·学习
桐盛科技9 分钟前
什么是边缘计算网关?桐盛科技拆解智慧公厕与能碳管理选型指南
人工智能·边缘计算
lailai041011 分钟前
AI远程控制的技术原理与实现路径
大数据·人工智能
科创致远11 分钟前
芜湖新能源汽车配套企业的多工厂命题:复制一家工厂,复制的是标准esop
大数据·人工智能·汽车·精益工程
陈彬深大23 分钟前
《AI 渐进编程》之二十七:抽象之后,精确性去了哪里?
人工智能