加速yolo3检测速度的两个思路

1.裁剪卷积核通道的方法来加速前向传播
https://github.com/Lam1360/YOLOv3-model-pruning

2.替换backbone网络,或者自己裁剪现有网络

步骤:
2.1 根据不同的model,要对已有的weights进行转换,得到与.cfg文件相匹配的初始权重。 partial命令可以分割权重文件
提取1~39层darknent19的参数,或者根据自己的需求来截取浅层网络对应的权重,保存成文件。

./darknet partial cfg/darknet19.cfg darknet19.weights darknet19.conv.40 40

2.2 tensorflow和keras可以通过指定层的名字来选择性加载权重(只加载一些自己感兴趣的),再进行训练。

  • 1
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值