yolov4-pytorch训练模型流程

1:下载labelimg软件 对要训练的图片进行标定生成XML 主意文件名最好重命名为序列

2:修改voc——annotation.py文件中分类的种类 classes=[]

3:使用voc2yolo4.py文件生成train.txt

4:train.py 需要修改的:

          (1)input_shape=[416,416] 如果现存够大可以608,608

          (2)consine_lr余弦退火衰减开关

        (3)mosaic 马赛克增强

        (4)cuda 是否开启CUDA GPU

        (5)smouth——labe原话 0.01 0.001

          (6)classes_paty=训练的类的名字的文件 

            (7)model_path 训练用权重模块

5:train.py开始训练

6:YOLO.py  修改model_path,model_image_size 与classes_path指向的train.py中设置的classes_path,input_shap的参数和生成的pth文件对应 

 

 

打包 安装pyinstaller(打包工具)
pip install pyinstaller

执行下面命令打包:
pyinstaller -F   XXXX.py

现阶段暂时打包不能用 因为有些资源无法被引入 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值