加速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可以通过指定层的名字来选择性加载权重(只加载一些自己感兴趣的),再进行训练。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章