YOLOX 复现

显卡 1660

环境windows10

mirrors / Megvii-BaseDetection / YOLOX · GitCode下载源码mirrors / Megvii-BaseDetection / YOLOX · GitCode

下载权重   

创建环境 

 

 这几个地方对好就行了。

打开终端输入

pip install -r requirements.txt

 python setup.py install

 下载apex

mirrors / nvidia / apex · GitCode

cd到apex-master，
python setup.py install

 测试  

没有保错

 右键运行出现下面这些就完成了

 新建文件夹放权重

 python tools/demo.py image -n yolox-s -c E:\YOLOX-main\w\yolox_s.pth --path assets/dog.jpg --conf 0.25 --nms 0.45 --tsize 640 --save_result --device gpu

 

出现这个错误解决

pip uninstall  torch

pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

python tools/demo.py image -n yolox-m -c E:\YOLOX-main\w\yolox_m.pth --path assets/dog.jpg --conf 0.25 --nms 0.45 --tsize 640 --save_result --device gpu

出现这个显卡内存不足

换模型或者  改变图片大小

这个地方要对着 

python tools/demo.py image -n yolox-s -c E:\YOLOX-main\w\yolox_s.pth --path assets/dog.jpg --conf 0.25 --nms 0.45 --tsize 320 --save_result --device gpu 

图片最后保存在