YOLOX自定义数据训练

有点个人笔记的意思,写的不是特别详细

  1. 数据转成voc格式ln到dataset目录下面;
  2. exps/example/yolox_voc/yolox_voc_s.py下修改self.num_classe
  3. data/datasets/voc_classes.py修改class name
  4. yolox/exp/yolox_base.py下修改epoch值和train_size,数据较少,epoch可以改小,train_size和自己数据集大小保持一致;
  5. 下载预训练模型yolox_m.pth
  6. 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

注意:

  1. 第六步可以改成其他模型
  2. train.py可以把一些默认参数写进去,不用训练时传参数
  3. 如果存储坐标是float类型,需要在voc.py修改一行
cur_pt = int(float(bbox.find(pt).text)) - 1

4.如果自定义数据size需要修改,最好height和width都修改为4的整数倍,否则会报错。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值