一种基于轻量级图像金字塔网络的单阶段物体检测方法

摘要

本发明涉及一种基于轻量级图像金字塔网络的单阶段物体检测方法，包括下列步骤:选定合适的物体检测的数据集，数据集包含图片和标注，标注含有检测框的位置和物体类别；构建用于单阶段物体检测的三个网络：提取多尺度深层特征的主干网络、用于对检测框分类、回归的分类回归子网络和所提出的轻量级图像金字塔网络；选择合适的损失函数，物体检测任务中设计为多任务的损失函数，包括分类损失函数和检测框回归的损失函数，设计网络训练循环迭代的次数和网络最终收敛条件，并初始化网络参数；从数据集里选取训练图像，进行计算和训练。