有点个人笔记的意思,写的不是特别详细
- 数据转成voc格式ln到dataset目录下面;
- exps/example/yolox_voc/yolox_voc_s.py下修改self.num_classe
- data/datasets/voc_classes.py修改class name
- yolox/exp/yolox_base.py下修改epoch值和train_size,数据较少,epoch可以改小,train_size和自己数据集大小保持一致;
- 下载预训练模型yolox_m.pth
- pip install -v -e .安装后执行python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 0 -b 16 --fp16 -o -c /path/to/yolox_m.pth
注意:
- 第六步可以改成其他模型
- train.py可以把一些默认参数写进去,不用训练时传参数
- 如果存储坐标是float类型,需要在voc.py修改一行
cur_pt = int(float(bbox.find(pt).text)) - 1
4.如果自定义数据size需要修改,最好height和width都修改为4的整数倍,否则会报错。