适于目标检测的MB-SSD方法和MB-SSD特征提取网络

摘要

本发明公开了一种适于目标检测的MB‑SSD方法，包括：提取多小目标图像，采用生成对抗网络对提取的多小目标图像进行增强；构建MB‑SSD特征提取网络，所述MB‑SSD特征提取网络包括主支特征提取网络、分支特征提取网络和定位网络，将增强后数据输入MB‑SSD特征提取网络，分别获取主支特征提取网络和分支特征提取网络的分类定位结果；再根据定位网络上同区域不同候选框的IoU重合度调整分类网络的输出特征；将主支特征提取网络和分支特征提取网络的分类定位结果进行融合并降维；计算模型损失，训练模型，优化模型参数。本发明能够在分类层中加入相对重合度调整分类结果的方法，来提高SSD算法的分类效果，同时有效的改善了对小目标的检测精度。