[发明专利]目标检测网络训练及检测方法、装置、终端及存储介质在审
| 申请号: | 202111624203.4 | 申请日: | 2021-12-28 |
| 公开(公告)号: | CN114462487A | 公开(公告)日: | 2022-05-10 |
| 发明(设计)人: | 蔡文强;叶丹;熊剑平;毛礼建 | 申请(专利权)人: | 浙江大华技术股份有限公司 |
| 主分类号: | G06K9/62 | 分类号: | G06K9/62;G06N3/04;G06N3/08;G06V10/764;G06V10/774;G06V10/80;G06V10/82 |
| 代理公司: | 深圳市威世博知识产权代理事务所(普通合伙) 44280 | 代理人: | 何倚雯 |
| 地址: | 310051 浙江*** | 国省代码: | 浙江;33 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 目标 检测 网络 训练 方法 装置 终端 存储 介质 | ||
1.一种目标检测网络训练方法,其特征在于,所述目标检测网络训练方法包括:
获取源域图像集和目标域图像集,所述源域图像集包括多张标注有目标真实信息以及第一真实域分类标签和第二真实域分类标签的第一扫描图像;所述目标域图像集包括多张标注有第一真实域分类标签和第二真实域分类标签且未标注目标信息的第二扫描图像;所述第一扫描图像和所述第二扫描图像分别基于不同的射线源扫描;
通过构建的初始检测网络对所述第一扫描图像和所述第二扫描图像分别进行检测,得到所述第一扫描图像的目标预测信息、第一预测域分类标签、第二预测域分类标签以及所述第二扫描图像的第一预测域分类标签和第二预测域分类标签;
基于同一所述第一扫描图像的目标预测信息与标注的所述目标真实信息、同一所述第一扫描图像的所述第一预测域分类标签与标注的所述第一真实域分类标签和所述第二预测域分类标签与标注的所述第二真实域分类标签以及同一所述第二扫描图像的所述第一预测域分类标签与标注的所述第一真实域分类标签和所述第二预测域分类标签与标注的所述第二真实域分类标签标签构建损失函数;
基于所述损失函数对所述初始检测网络进行迭代训练,得到目标检测网络。
2.根据权利要求1所述的目标检测网络训练方法,其特征在于,
构建所述初始检测网络的步骤包括:
基于初始目标检测网络、初始图像特征域分类网络和初始目标特征域分类网络构建所述初始检测网络;
其中,所述初始目标检测网络包括顺次连接的初始特征提取模块、初始特征聚合模块和初始目标预测模块;所述初始特征提取模块与所述初始图像特征域分类网络连接,所述初始特征聚合模块与所述初始目标特征域分类网络连接。
3.根据权利要求2所述的目标检测网络训练方法,其特征在于,
所述基于所述损失函数对所述初始检测网络进行迭代训练,得到目标检测网络,包括:
基于所述损失函数对所述初始检测网络中的初始目标检测网络、初始图像特征域分类网络和初始目标特征域分类网络中的权重进行修正,得到目标检测网络、图像特征域分类网络和目标特征域分类网络;
去除所述图像特征域分类网络和所述目标特征域分类网络,保留所述目标检测网络。
4.根据权利要求1所述的目标检测网络训练方法,其特征在于,所述第一扫描图像和所述第二扫描图像为不同射线源的X射线对应的扫描图像;所述目标预测信息包括目标预测位置和目标预测类别;
所述通过构建的初始检测网络对所述第一扫描图像和所述第二扫描图像分别进行检测,得到所述第一扫描图像的目标预测信息、第一预测域分类标签、第二预测域分类标签以及所述第二扫描图像的第一预测域分类标签和第二预测域分类标签,之前还包括:
将所述第一扫描图像和所述第二扫描图像的尺寸调整至第一预设尺寸;
并对所述第一扫描图像和所述第二扫描图像进行归一化处理,得到对应的预处理扫描图像。
5.根据权利要求2所述的目标检测网络训练方法,其特征在于,所述初始特征提取模块与所述初始图像特征域分类网络之间设置有第一梯度反转层;和/或所述初始特征聚合模块与所述初始目标特征域分类网络之间设置有第二梯度反转层。
6.根据权利要求4所述的目标检测网络训练方法,其特征在于,
所述通过构建的初始检测网络对所述第一扫描图像和所述第二扫描图像分别进行检测,得到所述第一扫描图像的目标预测信息、第一预测域分类标签、第二预测域分类标签以及所述第二扫描图像的第一预测域分类标签和第二预测域分类标签,包括:
通过所述初始特征提取模块对所述预处理扫描图像进行特征提取,得到对应的图像特征;
通过所述初始图像特征域分类网络对所述图像特征进行域适应,得到所述预处理扫描图像的第一预测域分类标签。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于浙江大华技术股份有限公司,未经浙江大华技术股份有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202111624203.4/1.html,转载请声明来源钻瓜专利网。





