加速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可以通過指定層的名字來選擇性加載權重(只加載一些自己感興趣的),再進行訓練。

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