模型训练方法、图像处理方法、装置和存储介质

    公开(公告)号:CN117333698A

    公开(公告)日:2024-01-02

    申请号:CN202311257276.3

    申请日:2023-09-27

    Abstract: 本公开提供了一种模型训练方法、图像处理方法、装置和存储介质,涉及图像处理领域。该方法包括:获取特征金字塔网络模型中样本图像的第i层的图像特征和第i+1层的融合特征;对第i层的图像特征对应的第一特征进行通道分组,得到多个第一子特征组,并提取每一个第一子特征组的全局特征;对第i+1层的融合特征进行上采样和卷积处理,得到第i+1层中间特征;根据每一个第一子特征组的全局特征与第i+1层中间特征的相关性,得到第i层的图像特征对应的特征精炼模块的注意力图;以及基于注意力图和注意力图对应的标签,对特征精炼模块进行训练。本公开在保证检测速率的同时,大幅度提高模型对多尺度目标的检测精度。

Patent Agency Ranking