1.conda create -n yolox python=3.7
2.conda activate yolox
3.pip install -r requirements.txt(会报错)
所以选择去pytorch官网下载其他版本:pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
4。之后再运行pip install -r requirements.txt即可
5.需要再次安装yolox(否则会报错):
python setup.py install
6.之后下载benchmark后面对应的weights放在/home/。。/YOLOX/yolox/models/下(还未下完)。
7.cd YOLOX后,运行python tools/demo.py image -n yolox-s -c yolox/models/yolox_s.pth --path assets/dog.jpg --conf 0.25 --nms 0.45 --tsize 640 --save_result --device gpu
之后就可以查看结果了(./YOLOX_outputs/yolox_s/vis_res/2023_05_14_15_44_28/dog.jpg):
8.也可以测视频,将视频放置在YOLOX/datasets/shipin2.mp4下,然后运行python tools/demo.py video -n yolox-s -c /home/liufengxiang/YOLOX/yolox/models/yolox_s.pth --path datasets/shipin2.mp4 --conf 0.25 --nms 0.45 --tsize 640 --save_result --device gpu
运行后就会yolox-output中有标记后的结果:
,