ssd网络forward理解

１．网络结构

主干网络：vgg + extras
分支网络：conf与loc网络

2．数据流

ssd中6层关键feature_map层，其feature_map的shape如下:

• n, 512, 38, 38
• n, 1024, 19, 19
• n, 512, 10, 10
• n, 256, 5, 5
• n, 256, 3, 3
• n, 256, 1, 1

分别经过了loc和conf层后, shape成为了

• loc-----------------------------------------------------------conf
• n, 16, 38, 38 --------------------------------------n, 84, 38, 38
• n, 24, 19, 19---------------------------------------n, 126, 38, 38
• n, 24, 10, 10---------------------------------------n, 126, 38, 38
• n, 24, 5, 5-------------------------------------------n, 126, 38, 38
• n, 16, 3, 3-------------------------------------------n, 84, 38, 38
• n, 16, 1, 1-------------------------------------------n, 84, 38, 38

分别拉平concate起来, 其shape分别为

loc_data : n x 34928 --------> n x 8732 x 4 -------- >4个坐标
conf_data : n x 183372-------> n x 8732 x 21-------->21个分类
(ssd一定要记住 8732 这个数字, 别问我为什么, 这是 先验框个数 )

后续操作

根据conf_data中的置信度进行阈值筛选, decode_boxes(即根据先验框与loc_data,找到真实框坐标)
再用nms去重