YOLOX系列文章(二):win10+cpu运行YOLOX

win10+cpu运行YOLOX教程

    • 0.本次任务
    • 1.搭建YOLOX开发环境
      • 下载文件
      • 安装win10+cpu版本所需的包
    • 2.win10+cpu运行YOLOX
    • 3.个人总结

0.本次任务

  1. 搭建YOLOX开发环境(对原始开发环境做稍许删减)
  2. win10+cpu正常运行
  3. 个人总结

1.搭建YOLOX开发环境

下载文件

下载YOLOX的github的代码,或是用git clone命令,或者点击下载,或是转到gitee上再下载,你随意。
YOLOX系列文章(二):win10+cpu运行YOLOX_第1张图片

安装win10+cpu版本所需的包

首先在anaconda中将隔离环境搭建好(搭建教程看这里),然后安装如下包:

  • python>=3.6
  • pytorch>=1.3
  • torch torchvision Pillow
  • opencv_python
  • loguru
  • scikit-image
  • tqdm
  • thop
  • tabulate
  • pycocotools
  • matplotlib

小福利(复制粘贴):
loguru安装:

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

tabulate安装:

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

pycocotools安装:

pip install cython -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install git+https://gitee.com/jiangjiajun/philferriere-cocoapi.git#subdirectory=PythonAPI

2.win10+cpu运行YOLOX

在pycharm中点击箭头所指的terminal,进入到Terminal下
YOLOX系列文章(二):win10+cpu运行YOLOX_第2张图片
然后按照YOLOX中github上的命令输入

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

你会发现提示如下错误ModuleNotFoundError: No module named ‘yolox’:
在这里插入图片描述

可以在from yolox.data.data_augment import preproc上面添加如下两行代码(注意路径替换成你自己的):

import sys
sys.path.append(r'D:\myProjects\YOLOX')

YOLOX系列文章(二):win10+cpu运行YOLOX_第3张图片

然后需要将官方给的命令行自定义,示例如下:

python tools/demo.py video -n yolox-s -c tools/yolox_s.pth --path tools/demonew.mp4 --conf 0.3 --nms 0.65 --tsize 640 --save_result --device cpu

从github上下载yolox_s.pth文件,放在tools文件夹下(急着和妹纸去打球,随便放位置了),下载一个mp4文件,放在tools下面,然后运行上面的命令,就等啊等。。。
在这里插入图片描述
然后就输出个文件,注意看上图,图中提示video path is …,你肯定会找得到的,效果如下(不知如何传视频 ,就截张图):

3.个人总结

1.在win10+cpu上运行yolox,感觉速度还可以,但是规范点是应该量化的
2.后面尝试下openvino,毕竟前几天参加活动,中了个VPU(手动狗头)

你可能感兴趣的