SSD Single shot detection 单发多框检测 实现 多尺度锚框 动手学深度学习v2

Posted 2022-05-03 AI架构师易筋

1. 多尺度锚框

2. SSD Single shot detection 单发多框检测 实现

num_classes + 1 表示分类数 + 背景数



channel先挪到最后

bbox_masks 表示如果是背景框的时候，是0，否则是1.只关心非背景的边框。

3. Q&A

• 1. num_inputs 输入的通道数，num_anchors每个像素的锚框数。 w*h*num_anchors*(num_class+1)
• 2. 像素比较大图片，SSD会炸掉，用YOLO实现。
• 3. 尺度特殊的物体，比如电线杠，要用YOLO检测。
• 4. CrossEntropy loss 和 L1 loss 的取值要加不同的权重。

参考

https://www.bilibili.com/video/BV1ZX4y1c7Sw?p=1