[发明专利]运算方法有效
申请号: | 201911406583.7 | 申请日: | 2018-12-29 |
公开(公告)号: | CN111193916B | 公开(公告)日: | 2022-03-29 |
发明(设计)人: | 不公告发明人 | 申请(专利权)人: | 中科寒武纪科技股份有限公司 |
主分类号: | H04N9/64 | 分类号: | H04N9/64;H04N9/68 |
代理公司: | 暂无信息 | 代理人: | 暂无信息 |
地址: | 100190 北京市海*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 运算 方法 | ||
1.一种运算方法,其特征在于,所述方法应用于异构计算架构中,所述异构计算架构包括通用处理器和人工智能处理器,包括:
在接收到待处理任务时,判断所述待处理任务的输入图像数据的第一格式与预设的Caffe图像处理模型所支持的输入数据的第二格式是否一致;
在所述第一格式与所述第二格式不一致时,根据所述第一格式及所述第二格式,调整所述Caffe图像处理模型的模型定义文件,以使根据调整后的所述模型定义文件生成的Caffe图像处理模型所支持的输入图像数据为所述第一格式,
其中,所述方法还包括:
在所述第一格式和所述第二格式为三基色图像数据格式、且在所述第二格式的通道个数少于所述第一格式的通道个数,且所述第二格式的通道顺序与所述第一格式的通道顺序不同时,根据第一格式与所述第二格式的通道个数以及通道顺序,调整所述模型定义文件,
其中,根据第一格式与所述第二格式的通道个数以及通道顺序,调整所述模型定义文件,包括:
调整所述模型定义文件中的首层卷积层对应的卷积核的通道顺序,并在所述首层卷积层对应的卷积核中添加卷积核权重为零的卷积通道,以使修改后的所述首层卷积层支持所述第一格式的输入图像数据。
2.根据权利要求1所述方法,其特征在于,在所述第二格式的通道个数等于所述第一格式的通道个数,且所述第一格式的通道顺序与所述第二格式的通道顺序不同时,所述根据第一格式与所述第二格式的通道个数以及通道顺序,调整所述模型定义文件,包括:
调整所述模型定义文件中的首层卷积层对应的卷积核的通道顺序,以使调整后的所述模型定义文件中的首层卷积层支持所述第一格式的输入图像数据。
3.根据权利要求1所述方法,其特征在于,在所述第二格式的通道个数大于所述第一格式的通道个数,且所述第二格式的通道顺序与所述第一格式的通道顺序相同,且所述第二格式比所述第一格式多余的通道权重为零时,所述根据第一格式与所述第二格式的通道个数以及通道顺序,调整所述模型定义文件,包括:
将所述模型定义文件中的首层卷积层对应的卷积核的权重为零的卷积通道删除,以使调整后的所述模型定义文件中的首层卷积层支持所述第一格式的输入图像数据。
4.根据权利要求1所述方法,其特征在于,在所述第一格式为亮度-色度图像数据格式,所述第二格式为三基色图像数据格式时,所述根据所述第一格式及所述第二格式,修改所述模型定义文件,包括:
在所述模型定义文件中添加第一数据转换层,所述第一数据转换层位于首层卷积层之前,用于将所述第一格式的输入图像数据转换为所述第二格式。
5.根据权利要求1所述方法,其特征在于,在所述第一格式为亮度-色度图像数据格式,所述第二格式为三基色图像数据格式时,所述根据所述第一格式及所述第二格式,修改所述模型定义文件,包括:
在所述模型定义文件中添加第二数据转换层,所述第二数据转换层用于将所述第一格式的输入图像数据转换为第三格式;
根据所述第三格式,修改所述模型定义文件中的首层卷积层,以使所述首层卷积层支持所述第三格式的输入图像数据,
其中,所述第三格式为四通道数据格式,四通道包括三原色通道和一个额外的透明度通道。
6.根据权利要求5所述方法,其特征在于,所述第二数据转换层,位于所述Caffe图像处理模型的首层卷积层之前。
7.根据权利要求5所述方法,其特征在于,根据所述第三格式,修改所述模型定义文件中的首层卷积层,包括:
调整所述首层卷积层对应的卷积核的通道顺序,并在所述卷积核中添加卷积核权重为零的卷积通道,以使调整后的所述首层卷积层所支持的输入数据为所述第三格式。
8.根据权利要求1-7中任意一项所述方法,其特征在于,所述方法还包括:
在接收到所述待处理任务的输入图像数据时,根据调整后的模型定义文件和权重文件,生成Caffe图像处理模型;
将所述输入图像数据输入生成的Caffe图像处理模型中处理,得到图像处理结果。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于中科寒武纪科技股份有限公司,未经中科寒武纪科技股份有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201911406583.7/1.html,转载请声明来源钻瓜专利网。