1、将训练好的权重文件放入权重文件夹
2、修改yolox/data/datasets/下的init.py文件,添加:from .voc_classes import VOC_CLASSES
3、toos/demo.py文件中将COCO_CLASSES全部修改为VOC_CLASSES
4、预测
若想进行视频预测,只需将image 更换为 video;
若想预测整个文件夹,将xxxx.jpg去掉,只留 --path assets/
python tools/demo.py image -f exps/example/yolox_voc/yolox_voc_s.py -c weights/best_ckpt.pth --path assets/class01.jpg --conf 0.25 --nms 0.45 --tsize 640 --save_result --device [cpu/gpu]