[发明专利]图像目标检测方法、装置、设备和存储介质

权利要求书

1.一种图像目标检测方法，其特征在于，包括：

获取待检测图像并将所述待检测图像输入训练好的图像检测模型，所述图像检测模型包括：特征提取模型、特征增强模型和检测头预测模型，所述特征提取模型包括：编码模块和解码模块；

利用所述编码模块对所述待检测图像进行特征提取得到至少一个编码特征图；

利用所述解码模块对至少一个所述编码特征图进行密集连接得到解码特征图；

利用所述特征增强模型对所述解码特征图进行特征金字塔变换得到增强特征图；

利用所述检测头预测模型对所述增强特征图进行预测得到目标检测结果。

2.根据权利要求1所述的图像目标检测方法，其特征在于，所述利用所述解码模块对至少一个所述编码特征图进行密集连接得到解码特征图，包括：

根据第一编码特征图的维度，对至少一个所述编码特征图进行特征图跳接得到第一解码特征图；

根据第二解码特征图的维度，对所述第一解码特征图和至少一个所述编码特征图进行特征图跳接得到所述第二解码特征图；

根据所述编码特征图的维度，对所述第二解码特征图和至少一个所述编码特征图进行特征图跳接得到所述解码特征图。

3.根据权利要求2所述的图像目标检测方法，其特征在于，至少一个所述特征图包括：第一编码特征图、第二编码特征图、第三编码特征图和第四编码特征图，所述根据所述第一编码特征图的维度，对至少一个所述编码特征图进行特征图跳接得到第一解码特征图，包括：

对所述第一编码特征图进行下采样，得到第一下采样结果；

对所述第二编码特征图进行下采样，得到第二下采样结果；

对所述第四编码特征图进行反卷积，得到第四反卷积结果；

利用预设第一权重参数集中各权重参数分别于与对应的所述第一下采样结果、所述第三编码特征图、所述第二下采样结果和所述第四反卷积结果相乘后相加，得到所述第一解码特征图。

4.根据权利要求3所述的图像目标检测方法，其特征在于，所述第二解码特征图的维度，对所述第一解码特征图和至少一个所述编码特征图进行特征图跳接得到所述第二解码特征图，包括：

对所述第一解码特征图进行反卷积，得到第一反卷积结果；

对所述第三编码特征图进行上采样，得到第三上采样结果；

利用预设第二权重参数集中各权重参数分别与对应的所述第一反卷积结果、所述第二编码特征图、所述第一下采样结果和所述第三上采样结果相乘后相加，得到所述第二解码特征图。

5.根据权利要求4所述的图像目标检测方法，其特征在于，所述根据所述编码特征图的维度，对所述第二解码特征图和至少一个所述编码特征图进行特征图跳接得到所述解码特征图，包括：

对所述第二解码特征图进行反卷积，得到第二反卷积结果；

对所述第二编码特征图进行上采样，得到第二上采样结果；

利用预设第三权重参数集中各权重参数分别与对应的所述第二反卷积结果、所述第一编码特征图、所述第二上采样结果和所述第三上采样结果相乘后相加，得到所述解码特征图。

6.根据权利要求1所述的图像目标检测方法，其特征在于，所述利用所述特征增强模型对所述解码特征图进行特征金字塔变换得到增强特征图，包括：

对所述解码特征图进行压缩得到至少一个压缩特征图；

对至少一个所述压缩特征图进行卷积得到至少一个压缩卷积特征图；

根据所述增强特征图的维度，对至少一个所述压缩卷积特征图进行对齐和上采样，得到所述增强特征图。