fully-connected轉conv

fully-connected轉conv 具體操作 假設fc輸入(h,w,c),fc輸出爲(fc_out),那麼fc的參數shape爲(h*w*c, fc_out)。通過使conv得kernel覆蓋fc的輸入,即conv的kernel_size = h*w,kernel_num = fc_out,則conv輸出shape爲(1,1,fc_out)。通過這種方式fc和conv其實是可以轉換的,如上c
相關文章
相關標籤/搜索