[发明专利]目标检测方法及装置

摘要

本发明公开了一种目标检测方法及装置，属于机器视觉领域。该方法包括：从第一CNN模型的第一浅卷积层的特征图中以边长小于或等于预设阈值的锚点框为基准回归出多个第一检测目标候选框；对该第一深卷积层的特征图进行全尺度目标候选框提取，得到多个第二检测目标候选框；从该多个第二检测目标候选框中选取边长小于或等于该预设阈值的多个第三检测目标候选框；基于该多个第一检测目标候选框和该多个第三检测目标候选框的重合度，从该多个第一检测目标候选框中选取多个第四检测目标候选框；基于该多个第四检测目标候选框，对该目标图像中的各个目标进行分类，得到各个目标属于预设类别的概率。本发明提高了对小尺寸目标进行检测和识别的准确度。

主权项

1.一种目标检测方法，其特征在于，所述方法包括：基于第一卷积神经网络CNN模型和待检测的目标图像，生成第一浅卷积层的特征图，所述第一浅卷积层是指所述第一CNN模型包括的多个卷积层中第一深卷积层之前的任一卷积层，所述第一深卷积层是指最靠近输出层的卷积层；从所述第一浅卷积层的特征图中以边长小于或等于预设阈值的锚点框为基准回归出目标候选框，得到多个第一检测目标候选框；基于所述第一CNN模型和所述第一浅卷积层的特征图，生成所述第一深卷积层的特征图；对所述第一深卷积层的特征图进行全尺度目标候选框提取，得到多个不同尺度的第二检测目标候选框；从所述多个不同尺度的第二检测目标候选框中选取边长小于或等于所述预设阈值的目标候选框，得到多个第三检测目标候选框；基于所述多个第一检测目标候选框和所述多个第三检测目标候选框的重合度，从所述多个第一检测目标候选框中选取多个第四检测目标候选框；基于所述多个第四检测目标候选框，对所述目标图像中的各个目标进行分类，得到各个目标属于预设类别的概率。