Keras的TimeDistributed层

Keras的TimeDistributed层主要用途是在时间维度上进行全连接.

比如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个分类和回归.

posted @ 2019-12-31 21:04  MSTK  阅读(633)  评论(0编辑  收藏