關於深度可分離卷積可以減少計算量的問題

一、什麼是深度可分離卷積?深度可分離卷積的操作流程?

二、爲什麼深度可分離卷積的計算量比普通卷積的計算量要小?

例如:12*12*3的原始圖像用256個5*5*3的卷積核卷出8*8*256的結果(stride=1)

1. 使用原始的卷積方法的計算量:256*(5*5*3)*(8*8)=1228800

注:單通道5*5卷積核的計算量爲5*5,所以3通道的爲5*5*3;

卷積核要在原始圖像上移動8*8次(橫向8次,縱向8次)

2. 使用深度可分離卷積的計算量:3*(5*5*1)*(8*8)+256*(1*1*3)*(8*8)=53952

Step 1:先進行深度卷積,用3個5*5*1的卷積核將12*12*3的原始圖卷出8*8*3的結果,就是3*(5*5*1)*(8*8)=4800。

Step 2:然後對8*8*3的圖用256個1*1*3的卷積核卷出最終的8*8*256的結果,就是256*(1*1*3)*(8*8)=49152。

從這個例子中可以看出使用深度可分離卷積是可以大大減少計算量的。

 

 

 

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