Caffe中卷積的實現

方法:將卷積轉化爲矩陣相乘 兩步: (1)調用im2col函數,將圖像(feature map)轉化爲矩陣 (2)調用GEMM函數,做實際的運算 優點:邏輯簡單,易於實現 缺點:耗費大量內存 feature map每一個卷積位置(卷積核將要滑過的位置),展開成一個 C∗K∗K 維的行向量 繼續展開…… 按照這個方式展開完畢之後,得到一個矩陣: H∗W 行(因爲有 H∗W 個卷積位置), C∗K∗K
相關文章
相關標籤/搜索