目標檢測中的注意力機制
在計算機視覺中,使用特徵圖金子塔,融合上文特徵,增加深度,增加廣度使用複雜的特徵圖處理手段已經是常態,最近對注意力機制的使用也愈演愈烈。
結合最近的注意力機制學習:
- 空間位置軟注意力機制:使用sigmoid對map進行操作,之後再與原來的map進行元素相加,相乘,concat。
- 通道注意力機制:一般是參照SEnet,對通道進行加權操作。
- 結合位置空間和通道注意力機制:1和2並行,1和2序列,1和2同時應用。
- 與GAN結合的注意力機制。
- 與RNN結合的注意力機制。
- 各種手段的注意力機制:在領域, RPN分佈上進行注意力機制調節。
@2018年是注意力機制提到最多的一年,每篇論文都會提一下。
SENet
SENet 提出了通道間的注意力機制,通過兩個全連線層計算通道之間的重要程度。從而過濾掉不重要的通道值。
使用:
- global average pooling 。把二維特徵圖變成實數
- 使用兩個壓縮的全連線層進行通道權重的學習
- 使用門控sigmoid或者softmax
- 進行sclae,把每個權重加到特徵圖上
重要程式碼段:
def Squeeze_excitation_layer(self, input_x, out_dim, ratio, layer_name):
with tf.name_scope(layer_name) :
squeeze = Global_Average_Pooling(input_x)
# 進行全域性池化
excitation = Fully_connected(squeeze, units=out_dim / ratio, layer_name=layer_name+'_fully_connected1')
# 學習通道權重
excitation = Relu(excitation)
excitation = Fully_connected(excitation, units=out_dim, layer_name=layer_name+'_fully_connected2')
# 學習通道權重
excitation = Sigmoid(excitation)
# 閾值調節
excitation = tf.reshape(excitation, [-1,1,1,out_dim])
scale = input_x * excitation
# 把權重賦給各個通道
return scale
相關文章
- 目標檢測
- 九、目標檢測
- 目標檢測---教你利用yolov5訓練自己的目標檢測模型YOLO模型
- 計算機視覺中的注意力機制計算機視覺
- 目標檢測面面觀
- 目標檢測之SSD
- 目標檢測綜述
- 28-目標檢測
- 目標檢測實用中可以改進的方向
- 目標檢測:二維碼檢測方案
- 注意力機制----RNN中的self-attentionRNN
- 注意力機制
- SSD 目標檢測 Keras 版Keras
- 目標檢測發展方向
- 目標檢測之YOLO系列YOLO
- 【目標檢測】R-CNNCNN
- 深度學習中的注意力機制(Attention Model)深度學習
- 心跳檢測機制
- 深度學習之目標檢測與目標識別深度學習
- 目標檢測相關論文
- 【目標檢測】Bounding Box Regression
- 深度學習之目標檢測深度學習
- 目標檢測:Segmentation is All You Need ?Segmentation
- 目標檢測資料集分析
- 2018目標檢測
- 目標檢測(Object Detection)總覽Object
- 目標檢測之RetinaNetNaN
- Object Detection(目標檢測神文)Object
- 不帶Anchors和NMS的目標檢測
- 0-目標檢測模型的基礎模型
- 注意力機制在圖卷積中的應用卷積
- 目標檢測模型的評價標準-AP與mAP模型
- [AI開發]目標檢測之素材標註AI
- 做目標檢測,這一篇就夠了!2019最全目標檢測指南
- 目標檢測演算法學習演算法
- 目標檢測(一):LeNet-5
- 訓練一個目標檢測模型模型
- 運動目標檢測與跟蹤