yolov12部署
戳链接访问原论文论文地址
戳链接访问原代码代码地址
直接把源代码以ZIP的形式下载到本地,然后解压用IDE打开就可以了(这一步比较简单不过多介绍)
在IDE中打开可以看见一个README.md文件,这里有我们将yolov12部署本地的一些要求
这里我用的是pycharm,在终端中按照文章的要求新建一个虚拟环境
conda create -n yolov12 python=3.11
然后,激活这个名为yolov12的虚拟环境
conda activate yolov12
接着我们还要安装几个重要的东西
- flash_attn-2.7.3+cu11torch2.2cxx11abiFALSE-cp311-cp311-linux_x86_64.whl这个windows操作系统下的平替
- 与第一条对应的pytorch
pytorch下载
将pytorch中的这段代码复制进终端中就可以下载对应的版本了
随后我们需要把刚才下载好的flash-attention windows端whl包导入
找到下载的文件所在,复制绝对路径如:E:\yolov12-main\flash_attn-2.7.4%2Bcu126torch2.6.0cxx11abiFALSE-cp311-cp311-win_amd64.whl
然后,在终端中运行如下代码,就可以导入这个包
pip install E:\yolov12-main\flash_attn-2.7.4%2Bcu126torch2.6.0cxx11abiFALSE-cp311-cp311-win_amd64.whl
最后我们找到requirements.txt文件,将其中的一部分注释掉
将代码注释掉之后运行如下代码,安装requirements.txt中需要的库就可以
pip install -r requirements.txt
然后执行如下代码安装ultralytics
pip install -e .
做到这一步基本我们就都做完了,然后可以用一下这段推理代码来测试一下有没有部署成功
yolo predict model=yolo11n.pt source='https://ultralytics.com/images/bus.jpg'
部署完成之后大家就可以根据自己的需要来进行操作了
时间比较紧,有一些细微之处可能没有写到,有问题的小伙伴可以留言,我知道的都会回复