Keras的TimeDistributed層

Keras的TimeDistributed層主要用途是在時間維度上進行全鏈接.spa

好比Faster RCNN,1張圖生成了16個ROI,須要對每個ROI進行分類和迴歸,ROI的維度是7×7×512,長度和寬度都是7,512個通道,16個ROI的的維度是16×7×7×512,須要獲得16個分類和迴歸,這個時候就能夠使用TimeDistributed層了.一次性輸入16×7×7×512,進行全鏈接,至關於16個時間切片,對每一個7×7×512的切片分別進行全鏈接(Dense),最後就獲得16個分類和迴歸.ast

相關文章
相關標籤/搜索