Caffe BN層合併

非常感謝博主小麥草的技術支持!

----------------------------------------------------------------------------------------------------

1. 合併方式

https://blog.csdn.net/kangdi7547/article/details/81348254,參考博客

使用python-caffe進行合併

參考博客方式轉換類似resnet模型。

因爲conv->bn->scale->prelu這幾層,特徵維度是不變的,所以首先要將這幾層做一個”in-place”操作,即將這幾層的bottom、top改名爲相同的,如下圖:

可視化圖爲:

另外,模型中還有很多的bn+scale層不能合併的,因爲其位於Eltwise後,上面沒有卷積層,這些只能暫時保留。

2. 實驗驗證

結果準確度:經過將實驗圖片進行打印,合併BN層以後的結果輸出和未合併之前的完全一致。

速度提升:

本 地 —— i7,NVIDIA RTX-2060,112*112的人臉圖像,特徵提取時間大概從16ms下降到13ms,即速度提升大概有20%。

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