YOLOX环境搭建与运行效果
通过git clone git@github.com:Megvii-BaseDetection/YOLOX.git或者下载方式获得旷世在github上共享的工程文件。1. 虚拟环境创建通过如下命令在conda下创建用于工程YOLOX的虚拟环境,假设虚拟环境为yoloxconda create -n yolox python=3.8然后通过命令conda activate yolox进入yolox虚
通过git clone git@github.com:Megvii-BaseDetection/YOLOX.git或者下载方式获得旷世在github上共享的工程文件。
1. 虚拟环境创建
通过如下命令在conda下创建用于工程YOLOX的虚拟环境,假设虚拟环境为yolox
conda create -n yolox python=3.8
然后通过命令
conda activate yolox
进入yolox虚拟环境,接着下面的安装。
2. pycocotools的安装
在Linux环境下安装pycocotools只需要通过pip install pycocotools。但是在Windows上安装会出错,找不到相关的C++编译器,需要安装visual cpp build tools 2015,非常的麻烦。Windows一个最直接的方法采用如下命令安装:
pip install pycocotools-windows
3. 安装YOLOX工程需要的包
运行YOLOX工程需要安装必备包,必备的包在YOLOX工程的requirements.txt中,在虚拟环境中通过如下命令安装
pip install -r requirements.txt
其中必备包如下:
4. YOLOX的检测测试
检测图像或者视频,首先需要下载训练好的权重文件,YOLOX中训练好的权重文件为pth格式,保存在工程目录weights下,其中下载地址如下:
https://github.com/Megvii-BaseDetection/YOLOX
最后下载的权重pth文件存储在工程weights中如下所示:
(1) 摄像头检测
根据YOLOX官方给出的代码,需要执行工程中tools/demo.py才能进行目标检测。
python tools/demo.py webcam -n yolox-s -c ../weights/yolox_s.pth --camid 0 --conf 0.25 --nms 0.45 --tsize 416 --save_result
通过执行上面的命令来检测摄像头中的内容。其中
webcam:表示以摄像头进行检测;
-n:表示检测模型名称
-c:加载的权重pth文件
--camid:指定读取数据的摄像机id
--conf:设置执行度阈值
--nms:设置非极大值抑制阈值
--tsize:设置网络输入的尺寸大小
--save_result:保存检测的结果
在pycharm IDE中配置运行的参数,如下图
如果需要在IDE中显示检测结果,需要只需在demo.py文件的imageflow_demo函数中添加cv2.imshow,如下图所示
如果需要退出程序,只需要按Q键就可以退出,如果设置--save_result,则退出程序会保存检测视频。
截取摄像头中一帧图像的检测结果如下图所示
摄像头每帧保存的结果视频
默认名称为camera.mp4。
更多推荐
所有评论(0)