SSD算法学习及PyTorch代码分析[2]---Prior Box分析

本文深入解析SSD目标检测算法中的Prior Box层,探讨了如何生成不同尺度和宽高比的先验框,并通过PyTorch代码进行分析,包括尺度设置的线性增长规则和宽高比的设定,以及在不同特征图上的应用。
摘要由CSDN通过智能技术生成

上一篇讲了SSD算法的整体框架,这一篇分析SSD的 Prior Box 层,先验框是如何产生
SSD算法网络结构以及YOLO网络结构对比
SSDYOLO不同,每一个 feature map 上的点都为一个 cell, 而YOLO是划分为 gird 。
假如 feature map 大小为 m × n m \times n m×n 那麽会有 m × n m \times n m×n 个 cell 。每个 cell 会生成固定 scaleaspect ratio 的 box 。
假如每个 cell 生成 k k k 个 box , 每个 box 对应类 c c c 个类别以及 4 个坐标偏移量。因此输出的先验框就有 m × n × k × ( c + 4 ) m \times n \times k \times (c+4) m×n×k×(c+4)

因此这里有着三个超参数(scale, aspect ratio以及 k k k)。

1. Scale设置
scale 有着一个线性增长的规则:
s k = s m i n + s m a x − s m i n m − 1 × ( k − 1 ) s_{k} = s_{min} + \frac{ {s_{max}-s_{min}}}{m-1}\times(k-1) sk=smin+m1smaxsmin×(k1)
其中, k ∈ [ 1 , m ] k \in [1,m] k[1,m] s k s_{k}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值