[发明专利]神经网络模型的转换方法、模型精度定位方法及装置在审
申请号: | 202010129010.0 | 申请日: | 2020-02-28 |
公开(公告)号: | CN113326913A | 公开(公告)日: | 2021-08-31 |
发明(设计)人: | 罗培超;徐迟;张行程 | 申请(专利权)人: | 上海商汤智能科技有限公司 |
主分类号: | G06N3/04 | 分类号: | G06N3/04;G06N3/08 |
代理公司: | 广州三环专利商标代理有限公司 44202 | 代理人: | 熊永强;董文俊 |
地址: | 200233 上海市徐*** | 国省代码: | 上海;31 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 神经网络 模型 转换 方法 精度 定位 装置 | ||
1.一种神经网络模型的转换方法,其特征在于,所述方法包括:
模型转换装置获取源框架下的第一神经网络模型、模型输入数据以及目标框架的信息;
所述模型转换装置对所述源框架下的第一神经网络模型进行模型转换处理,得到所述目标框架下的第二神经网络模型;
所述模型转换装置基于所述第一神经网络模型和所述第二神经网络模型分别对所述模型输入数据进行处理得到的第一推理结果和第二推理结果,得到所述第二神经网络模型的精度验证结果。
2.根据权利要求1所述的方法,其特征在于,所述对所述源框架下的第一神经网络模型进行模型转换处理,得到所述目标框架下的第二神经网络模型,包括:
通过第一进程对所述源框架下的第一神经网络模型进行模型转换处理,得到所述目标框架下的第二神经网络模型;
所述方法还包括:
通过所述第一进程创建第二进程;
通过所述第一进程调用所述第二进程,以在所述第二进程中通过所述第二神经网络模型对所述模型输入数据进行处理,得到第二推理结果。
3.根据权利要求1或2所述的方法,其特征在于,所述模型转换装置基于所述第一神经网络模型和所述第二神经网络模型分别对所述模型输入数据进行处理得到的第一推理结果和第二推理结果,得到所述第二神经网络模型的精度验证结果,包括:
若所述第一推理结果与所述第二推理结果之间的差别处于预设误差范围内,确定所述第二神经网络模型的精度验证结果为精度对齐;和/或
若所述第一推理结果与所述第二推理结果之间的差别超出所述预设误差范围,确定所述第二神经网络模型的精度验证结果为精度未对齐。
4.根据权利要求1至3中任一项所述的方法,其特征在于,所述方法还包括:
响应于确定所述第二神经网络模型的精度验证结果为精度未对齐,所述模型转换装置定位所述第二神经网络模型中精度未对齐的目标网络层;
所述模型转换装置输出所述目标网络层的信息。
5.根据权利要求4所述的方法,其特征在于,所述定位所述第二神经网络模型中精度未对齐的目标网络层,包括:
所述模型转换装置建立所述第一神经网络模型和所述第二神经网络模型中的网络层之间的映射关系;
所述模型转换装置基于所述第一神经网络模型和所述第二神经网络模型分别处理所述模型输入数据的过程中每个网络层的输出数据以及所述映射关系,确定所述第二神经网络模型中精度未对齐的目标网络层。
6.根据权利要求5所述的方法,其特征在于,其中,所述模型转换装置对所述第一神经网络模型再次进行模型转换处理,其中,在所述再次进行模型转换处理的过程中,获取所述第一神经网络模型中的每个网络层的输出数据,和/或建立所述第一神经网络模型和所述第二神经网络模型中的网络层之间的映射关系。
7.根据权利要求1至6中任一项所述的方法,其特征在于,所述模型转换装置对所述源框架下的第一神经网络模型进行模型转换处理,得到所述目标框架下的第二神经网络模型,包括:
所述模型转换装置通过所述第一神经网络模型对所述模型输入数据的处理,建立网络拓扑图,其中,所述网络拓扑图包括所述第一神经网络模型中的网络层的信息以及网络层的连接关系;
所述模型转换装置基于所述网络拓扑图以及所述目标框架的信息,得到所述第二神经网络模型。
8.一种模型精度定位方法,其特征在于,包括:
模型精度定位装置获取源框架下的第一神经网络模型、模型输入数据以及目标框架的信息;
所述模型精度定位装置对所述第一神经网络模型进行模型转换处理,得到所述目标框架下的第二神经网络模型;
所述模型精度定位装置基于所述第一神经网络模型和所述第二神经网络模型分别处理所述模型输入数据的每个网络层的输出数据,定位所述第二神经网络模型中精度未对齐的目标网络层;
所述模型精度定位装置输出所述目标网络层的信息。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于上海商汤智能科技有限公司,未经上海商汤智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202010129010.0/1.html,转载请声明来源钻瓜专利网。
- 上一篇:基于区块链的数据处理方法及装置
- 下一篇:液冷马达散热结构