Non_Local 網絡模塊的解析

該模塊能夠插入到現有的CNN網絡結構中,優勢是能夠提升總體的性能,缺點是參數多,計算量大python 寫此文章只爲本身往後回顧網絡 首先base cnn 爲resnet50 ,插入在block3以後,模塊的輸入和輸出都不變,這也是能夠即插即用的緣由。app 首先輸入爲【B,H,W,C】 先對輸入使用1X1大小的卷積核作卷積,下降輸入通道。減小計算量。而後把【B,H,W,C】reshape 成[B,
相關文章
相關標籤/搜索