yolox 训练自己数据集

目录

1.训练

2.测试

---

1.训练

 python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 1 --fp16 -c weight/yolox_s.pth

-d 显卡号（我的0是集显1是独显）

-b 1（4以上报错RuntimeError: CUDA out of memory）

 

 https://github.com/Megvii-BaseDetection/YOLOX/issues/741

解决FileNotFoundError: [Errno 2] No such file or directory: '02804.xml'

2.测试

python tools/demo.py image -f exps/example/yolox_voc/yolox_voc_s.py -c YOLOX_outputs/yolox_voc_s/best_ckpt.pth  --conf 0.3 --nms 0.65 --tsize 640 --save_re
sult --device gpu