IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> Python知识库 -> Windows下YOLOX的配置 -> 正文阅读

[Python知识库]Windows下YOLOX的配置

Windows下YOLOX的配置

作者:肆十二

本次博客电脑基本配置情况:

系统-windows10 显卡-RTX2070s CPU-I7 9700k

YOLOX是是旷视科技在2021年提出的Anchor Free的目标检测框架,YOLOX-l模型在coco的测试集上可以达到51.5的mAP值,相对于YOLOV5有1.8个点的提高,并且YOLOX是一个Anchor free的框架,相对来说没有太多需要调试的超参数,在原始的预训练模型上做一下微调就可以在自己的数据集上达到不错的效果,算法的细节部分还请大家参考作者的论文:YOLOX: Exceeding YOLO Series in 2021,另外,论文的作者在知乎上亲自对论文做了解读,感兴趣的小伙伴也可以去看看:如何评价旷视开源的YOLOX,效果超过YOLOv5? - 知乎 (zhihu.com)

好!我们今天就来一起看看如何在windows下配置YOLOX,YOLOX的实现有两个版本,一个是Pytorch版本的,一个是MegEngine版本的,megengine是旷视自研的深度学习框架,使用的人相对较少,我们今天主要来看看Pytorch版本代码的配置。

前期的准备工作

下载代码之前,请大家先安装好anaconda、pycharm以及安装好显卡驱动,因为之前的博客中我详细介绍了这两个安装步骤,并且还录制了专门的视频,不清楚的小伙伴可以先看这两篇博客补充一下自己的基础知识。

如何在pycharm中配置anaconda的虚拟环境_dejahu的博客-CSDN博客_如何在pycharm中配置anaconda

2021年Windows下安装GPU版本的Tensorflow和Pytorch_dejahu的博客-CSDN博客

配置环境

  1. 建立虚拟环境

    在cmd中执行下面的指令建立一个python版本为3.7.3,名字为yolox_torch的虚拟环境

    conda create -n yolox_torch python==3.7.3
    

    等待依赖安装完成之后会出现下面的信息

    [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-FdeUtDbh-1633072567557)(C:\Users\Scm97\AppData\Roaming\Typora\typora-user-images\image-20210920151021620.png)]

    激活这个虚拟环境,激活之后前面的括号表示你当前所处的虚拟环境

    image-20210920151116827

  2. 安装pytorch

    为了方便后期训练使用,我们这里安装的是GPU版本的Pytorch,Pytorch的版本为1.8.0,torchvision的版本为0.9.0

    conda install pytorch==1.8.0 torchvision==0.9.1
    

    安装完成之后,大家请安装下图所示的步骤查看自己pytorch的GPU是否可用,如果最后输出的为True,表示GPU可用。

    image-20210920152247817

  3. 下载代码并安装

    首先需要从官网下载代码,请执行下面的指令或者是直接下载压缩包

    git clone git@github.com:MegEngine/YOLOX.git # 网络好的用这个
    git clone https://gitee.com/mirrors/YOLOX.git
    

    执行下面的指令安装yolox(注:因为我们提前安装了torch和torchvision,请大家注释掉requirements.txt3和8行)

    cd YOLOX
    pip install -r requirements.txt
    pip3 install -v -e .  # or  python3 setup.py develop
    

    出现下面的信息表示安装成功

    image-20210920152946673

    安装Pycocotools,这个工具将会帮助你加载coco格式的数据集进行训练和测试

    pip install cython
    pip install pycocotools-windows
    
  4. 下载官方提供的模型

    官方的文件中提供了预训练的模型,也就是在coco数据集上训练好的模型,可以检测大概80类,已经足够大家使用了

    这里一共提供了5个模型,YOLOX-s是参数量比较小的版本,我们下载这个版本的模型就足够了,点击下面的按钮下载。

    [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-uO6bcBO6-1633072567574)(C:\Users\Scm97\AppData\Roaming\Typora\typora-user-images\image-20210920153342332.png)]

    我们在代码根目录下新建一个pretrained目录,把下载好的模型放在这个目录下即可,如下所示:

一起看看效果吧

下面我们将会对目标检测中非常经典的车狗图进行检测

python tools/demo.py image -n yolox-s -c pretrained/yolox_s.pth --path assets/dog.jpg --conf 0.25 --nms 0.45 --tsize 640 --save_result --device gpu

执行的具体过程如下:

image-20210920154451562

然后我们在上面保存的目录打开图片看看吧!

dog

连垃圾桶都检测出来了,是不是非常不戳!

另外,如果是自己写的配置文件,请执行下面的指令

python tools/demo.py image -f exps/default/yolox_s.py -c /path/to/your/yolox_s.pth --path assets/dog.jpg --conf 0.25 --nms 0.45 --tsize 640 --save_result --device [cpu/gpu]

如果检测视频的话,请执行这样的指令

python tools/demo.py video -n yolox-s -c /path/to/your/yolox_s.pth --path /path/to/your/video --conf 0.25 --nms 0.45 --tsize 640 --save_result --device [cpu/gpu]

python tools/demo.py video -n yolox-s -c pretrained/yolox_s.pth --path assets/test.mp4 --conf 0.25 --nms 0.45 --tsize 640 --save_result --device gpu
  Python知识库 最新文章
Python中String模块
【Python】 14-CVS文件操作
python的panda库读写文件
使用Nordic的nrf52840实现蓝牙DFU过程
【Python学习记录】numpy数组用法整理
Python学习笔记
python字符串和列表
python如何从txt文件中解析出有效的数据
Python编程从入门到实践自学/3.1-3.2
python变量
上一篇文章      下一篇文章      查看所有文章
加:2021-10-02 14:37:41  更:2021-10-02 14:39:04 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/15 17:01:26-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码