适于目标检测的MB-SSD方法和MB-SSD特征提取网络
基本信息
申请号 | CN202010810771.2 | 申请日 | - |
公开(公告)号 | CN111680705B | 公开(公告)日 | 2021-02-26 |
申请公布号 | CN111680705B | 申请公布日 | 2021-02-26 |
分类号 | G06K9/46(2006.01)I;G06N3/04(2006.01)I;G06K9/62(2006.01)I;G06N3/08(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 夏景明;张宇;谈玲 | 申请(专利权)人 | 盐城芯丰微电子有限公司 |
代理机构 | 南京钟山专利代理有限公司 | 代理人 | 陈月菊 |
地址 | 224014 江苏省盐城市盐都区盐龙街道创新中心16楼1601室(D) | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种适于目标检测的MB‑SSD方法,包括:提取多小目标图像,采用生成对抗网络对提取的多小目标图像进行增强;构建MB‑SSD特征提取网络,所述MB‑SSD特征提取网络包括主支特征提取网络、分支特征提取网络和定位网络,将增强后数据输入MB‑SSD特征提取网络,分别获取主支特征提取网络和分支特征提取网络的分类定位结果;再根据定位网络上同区域不同候选框的IoU重合度调整分类网络的输出特征;将主支特征提取网络和分支特征提取网络的分类定位结果进行融合并降维;计算模型损失,训练模型,优化模型参数。本发明能够在分类层中加入相对重合度调整分类结果的方法,来提高SSD算法的分类效果,同时有效的改善了对小目标的检测精度。 |
