安装完YOLOX的环境后,需要验证一下环境是否可用,才能进行下一步工作
-
下载预训练模型,本文选用的是YOLOX-s,下载地址: https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_s.pth
python tools/demo.py image -f exps/default/yolox_s.py -c ./yolox_s.pth --path assets/dog.jpg --conf 0.3 --nms 0.65 --tsize 640 --save_result --device gpu
参数说明
-
c:权重的路径
-
path: 测试图片的路径
-
conf: 置信度阈值
-
nms :nms的IOU阈值
-
tsize :测试图片resize的大小
-
save_result: 是否保存推理结果
-
device: 选用gpu或cpu推理