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