本节学习了SSD目标检测算法,了解了SSD算法可以分为两类:
1.two-stage方法:RCNN系列
通过算法产生候选框,然后再对这些候选框进行分类和回归。
2.one-stage方法:YOLO和SSD
直接通过主干网络给出类别位置信息,不需要区域生成。
通常情况下,one-stage方法速度更快,two-stage方法精度更高。
模型结构
SSD采用VGG16作为基础模型,然后在VGG16的基础上新增了卷积层来获得更多的特征图以用于检测。
模型特点
- 多尺度检测,多尺度检测,最后检测效果更好,更容易检测出小目标。
- 采用卷积进行检测
- 预设anchor,预设了预测框,更容易训练。
模型构建
损失函数