一、资源获取
源码:https://github.com/Megvii-BaseDetection/YOLOX
原文:https://arxiv.org/abs/2107.08430
我的测试代码:下载链接: https://pan.baidu.com/s/1rX4iRLavFc9Uoc8_otFIoA
密码: gmic
所有文件、安装包都在里面,文件较大,可以按需下载。
二、测试步骤
- python最好大于3.5(我的3.5有部分库无法下载)
- pytorch下载最新的吧(我的1.5.0不太行,换成1.9.0的了)
# 先卸载原torch、torchvision再安装 pip install torch torchvision -i https://pypi.mirrors.ustc.edu.cn/simple/
- 按照requirements.txt安装其他基本库,缺啥补啥。
-
下载安装pycocotools(其他方法可能由于网络原因,会安装失败,可参考以下步骤)
(1)下载安装git:下载链接: 密码: kf65。。。一路默认安装即可。
(2)下载编译cocoapi:注意:修改删除cocoapi-master/PythonAPI/setup.py文件中extra_compile_args部分参数。修改结果如下:
然后,执行以下操作:ext_modules = [ Extension( 'pycocotools._mask', sources=['../common/maskApi.c', 'pycocotools/_mask.pyx'], include_dirs = [np.get_include(), '../common'], extra_compile_args=['-std=c99'], ) ]
(3)下载pycocotools:下载链接: 密码: ic6g。。。文件较大,请坚持下载完,耐心等待。cd cocoapi-master cd PythonAPI python setup.py build_ext --inplace python setup.py build_ext install
solved_packageMissInInstalling_mu_visual_cpp_build_tools_2015_update_3_x64_dvd_dfd9a39c.iso
,并安装。链接。。。。一路默认安装即可。 - 下载源码
# 测试时可能会提示yolox文件夹找不到 # 在tools/demo.py的import cv2下面添加几行 import sys sys.path.append('D:/zhankun/YOLOX-main') sys.path.append('D:/zhankun/YOLOX-main/yolox')
- 下载权重文件(放在根目录下YOLOX-main):下载链接
- 测试,在根目录YOLOX-main下执行以下命令:
python tools/demo.py image -f exps/default/yolox_s.py -c ./yolox_s.pth.tar --path assets/dog.jpg --conf 0.3 --nms 0.65 --tsize 640 --save_result --device cpu