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

相关文章
相关标签/搜索