[发明专利]模型训练方法及图像处理方法和相关装置、设备、介质在审

专利信息
申请号: 202210089853.1 申请日: 2022-01-25
公开(公告)号: CN114445677A 公开(公告)日: 2022-05-06
发明(设计)人: 胡志强 申请(专利权)人: 上海商汤智能科技有限公司
主分类号: G06V10/774 分类号: G06V10/774;G06V10/776;G06K9/62
代理公司: 深圳市威世博知识产权代理事务所(普通合伙) 44280 代理人: 何倚雯
地址: 200233 上海市徐*** 国省代码: 上海;31
权利要求书: 查看更多 说明书: 查看更多
摘要:
搜索关键词: 模型 训练 方法 图像 处理 相关 装置 设备 介质
【权利要求书】:

1.一种模型训练方法,其特征在于,包括:

利用第一勾画模型对第一样本图像进行勾画,得到所述第一勾画模型中第一目标网络层输出的第一样本特征图;

利用第二勾画模型对第二样本图像进行勾画,得到所述第二勾画模型中第二目标网络层输出的第二样本特征图;

利用配准模型对样本图像对进行分析,得到样本形变参数;其中,所述样本图像对包括所述第一样本图像和所述第二样本图像;

基于所述样本形变参数,对所述第一样本特征图进行形变,得到形变样本特征图;

基于所述形变样本特征图和所述第二样本特征图之间的差异,调整所述配准模型的网络参数。

2.根据权利要求1所述的方法,其特征在于,所述方法还包括:

选择所述第一勾画模型中若干网络层分别作为所述第一目标网络层,以及选择所述第二勾画模型中若干网络层分别作为所述第二目标网络层;

选择满足预设条件的所述第一目标网络层和所述第二目标网络层进行组合,得到若干个网络层组合;

所述基于所述样本形变参数,对所述第一样本特征图进行形变,得到形变样本特征图,包括:

对于各个所述网络层组合,利用所述样本形变参数对所述网络层组合中所述第一目标网络层所输出的第一样本特征图进行形变,得到与所述网络层组合对应的形变样本特征图;以及

所述基于所述形变样本特征图和所述第二样本特征图之间的差异,调整所述配准模型的网络参数,包括:

基于各个所述网络层组合中所述第二目标网络层所输出的第二样本特征图和参考样本特征图之间的差异,调整所述配准模型的网络参数;其中,所述参考样本特征图为与所述网络层组合对应的形变样本特征图。

3.根据权利要求2所述的方法,其特征在于,所述选择所述第一勾画模型中若干网络层分别作为所述第一目标网络层,包括:

在所述第一勾画模型中,选择网络层数大于第一阈值的网络层,作为所述第一目标网络层;以及,

所述选择所述第二勾画模型中若干网络层分别作为所述第二目标网络层,包括:

在所述第二勾画模型中,选择网络层数大于第二阈值的网络层,作为所述第二目标网络层。

4.根据权利要求2所述的方法,其特征在于,所述预设条件包括:所述第一目标网络层所输出的第一样本特征图和所述第二目标网络层所输出的第二样本特征图具有相同分辨率。

5.根据权利要求1所述的方法,其特征在于,所述第一样本图像和所述第二样本图像均包含目标对象,所述第一勾画模型勾画得到所述第一样本图像中所述目标对象的第一勾画结果,所述第二勾画模型勾画得到所述第二样本图像中所述目标对象的第二勾画结果,且所述勾画结果表示所述目标对象的预测轮廓;在所述基于所述形变样本特征图和所述第二样本特征图之间的差异,调整所述配准模型的网络参数之前,所述方法还包括:

利用所述样本形变参数对所述第一勾画结果进行形变,得到形变勾画结果,并基于所述形变勾画结果与所述第二勾画结果之间的差异,得到第一损失;以及

所述基于所述形变样本特征图和所述第二样本特征图之间的差异,调整所述配准模型的网络参数,包括:

基于所述形变样本特征图和所述第二样本特征图之间的差异,得到第二损失;

基于所述第一损失和所述第二损失,得到总损失;

基于所述总损失,调整所述配准模型、所述第一勾画模型和所述第二勾画模型的网络参数。

6.根据权利要求5所述的方法,其特征在于,所述第一样本图像具有第一标记结果,所述第二样本图像具有第二标记结果,且标记结果表示所述目标对象的样本轮廓;在所述基于所述第一损失和所述第二损失,得到总损失之前,所述方法还包括:

基于所述第一标记结果和所述第一勾画结果之间的差异,得到第三损失,并基于所述第二标记结果和所述第二勾画结果之间的差异,得到第四损失;

利用所述样本形变参数对所述第一样本图像进行形变,得到形变样本图像,并基于所述形变样本图像和所述第二样本图像之间的差异,得到第五损失;

所述基于所述第一损失和所述第二损失,得到总损失,包括:

基于所述第一损失、所述第二损失、所述第三损失、所述第四损失和所述第五损失,得到所述总损失。

下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。

该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于上海商汤智能科技有限公司,未经上海商汤智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服

本文链接:http://www.vipzhuanli.com/pat/books/202210089853.1/1.html,转载请声明来源钻瓜专利网。

同类专利
专利分类
×

专利文献下载

说明:

1、专利原文基于中国国家知识产权局专利说明书;

2、支持发明专利 、实用新型专利、外观设计专利(升级中);

3、专利数据每周两次同步更新,支持Adobe PDF格式;

4、内容包括专利技术的结构示意图流程工艺图技术构造图

5、已全新升级为极速版,下载速度显著提升!欢迎使用!

请您登陆后,进行下载,点击【登陆】 【注册】

关于我们 寻求报道 投稿须知 广告合作 版权声明 网站地图 友情链接 企业标识 联系我们

钻瓜专利网在线咨询

周一至周五 9:00-18:00

咨询在线客服咨询在线客服
tel code back_top