![](https://img-blog.csdnimg.cn/20191013150232137.jpg?x-oss-process=image/resize,m_fixed,h_224,w_224)
ssd源码解读
SSD(Single Shot MultiBox Detector)原理和 源码解读
ZJE_ANDY
这个作者很懒,什么都没留下…
展开
-
ssd(Single Shot MultiBox Detector)代码之(五) 训练自己的数据集
原始代码来源:https://github.com/amdegroot/ssd.pytorch我修改后的代码:https://github.com/Andy-zhujunwen/ssd-use-your-own-dataset-实验气球数据集:https://pan.baidu.com/s/13DxzPZD1MhL_1KrKq55rfw&shfl=sharepset提取码:ca...原创 2019-10-21 17:31:32 · 2340 阅读 · 1 评论 -
ssd(Single Shot MultiBox Detector)代码解读之(四) 非极大值抑制(nms)解读
首先简单介绍一下nms:简介:nms是用于消除框的,例如下图,一个目标中有多个候选框的时候,nms就可以把候选框删剩下一个:注意:nms是消除表示同一类别的框的。得到:步骤:那么nms的步骤为:(1)将所有框的得分排序,选中最高分及其对应的框(2)遍历其余的框,如果和当前最高分框的重叠面积(IOU)大于一定阈值,我们就将框删除(3)从未处理的框中继续选一...原创 2019-10-21 08:28:39 · 3644 阅读 · 2 评论 -
ssd(Single Shot MultiBox Detector)代码解读之(三)multibox loss损失函数
multiboxloss是SSD的损失函数跟交叉熵那些官方给出接口的损失函数不同。multiboxloss需要自己定义的。建议:边看代码边看此博客。代码来源:https://github.com/amdegroot/ssd.pytorchmultiboxloss的流程:1.获取ssd网络的输出ssd网络输出有三个,其中两个是预测值,一个是所有defaultbox...原创 2019-10-21 02:59:26 · 4409 阅读 · 4 评论 -
ssd(Single Shot MultiBox Detector)代码解读之(二)default box
要是对SSD暂时没上面了解的,可以先看看下面的链接文章:SSD网络解读defaultbox部分:画defaultbox框的原则是在六个特征图上的每个像素点都画4或6个框。所以只需要知道6个特征图大小就能得到8732个特征图的归一化信息。首先先看一部分pytorch的代码: def forward(self): mean = [] fo...原创 2019-10-18 11:43:15 · 906 阅读 · 0 评论 -
ssd(Single Shot MultiBox Detector)解读之(一)原理解析
代码来源:https://github.com/amdegroot/ssd.pytorchSSD简介:ssd有ssd300和ssd512,一般我们都是用ssd300,意思是输入SSD网络的图片的尺寸是300x300的。(一)SSD的网络结构:SSD的网络结构绝不是简单的VGG16网络,如下图:图1它是由VGG16的网络改造而成的,具体步骤如下:1.先把传统的VGG1...原创 2019-10-17 22:31:39 · 5006 阅读 · 4 评论