图像检测模型训练方法、装置及图像检测方法

    公开(公告)号:CN115331097A

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

    申请号:CN202210806731.X

    申请日:2022-07-08

    Inventor: 杨刚 王艺莎 卢昊

    Abstract: 本发明实施例提供一种图像检测模型训练方法、装置及图像检测方法,该图像检测模型训练包括:通过特征提取网络对样本图像进行特征提取,得到第一特征图;将第一特征图输入到图像级域分类器,根据第一特征图中各区域的注意力权重更新第一特征图,以得到第二特征图;将第二特征图输入实例检测网络,以得到各实例的位置区域和类别分类结果;根据第二特征图生成第三特征图;将第三特征图输入目标实例类别标签对应的目标实例级域分类器,以获得目标实例级域分类器输出的域类别预测结果;根据域类别预测结果以及类别分类结果、回归参数,训练检测网络,以得到用于对目标域的待检测图像进行实例检测的目标检测模型。

Patent Agency Ranking