一种基于复合主干网络和多预测头的航拍目标检测方法

    公开(公告)号:CN115035429A

    公开(公告)日:2022-09-09

    申请号:CN202210748203.3

    申请日:2022-06-29

    Abstract: 本发明公开了一种基于复合主干网络和多预测头的航拍目标检测方法,涉及计算机视觉目标检测技术领域,包括以下步骤:步骤1、准备航拍图像的数据集;步骤2、构建目标检测模型;步骤3、应用目标检测模型进行目标预测;其中,步骤2包括:步骤2.1、建立目标检测模型的主干网络,并提取多尺度特征;步骤2.2、通过Recursive‑FPN网络对多尺度特征进行融合,得到融合增强后的多尺度特征;步骤2.3、基于融合增强后的多尺度特征,使用TPH网络作为detection head,构建目标检测模型;步骤2.4、将航拍图像的数据集按比例划分为训练集和测试集,用训练集输入模型进行训练,再用测试集对模型进行测试,确定目标检测模型。该方法实现航拍图像目标检测速度与精度更好的权衡。

Patent Agency Ranking