官网地址代码:https://github.com/Megvii-BaseDetection/YOLOX
训练过程参考:深入浅出Yolox之自有数据集训练超详细教程
训练命令:
python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 0 -b 8 -c backup/yolox_s.pth
验证命令:
python tools/eval.py -f exps/examples/yolox_voc/yolox_voc_s.py -c /root/work/yolox_src-sign/YOLOX_outputs/yolox_voc_s/best_ckpt.pth -b 8 -d 1 --conf 0.001 [--fp16] [--fuse]
注意官网的训练和验证命令是针对COCO数据集的。
不论是eval.py还是train.py都能看到,读取exp是根据args.exp_file或者args.name来选择的。