[发明专利]网络结构的处理方法、装置及相关产品有效
| 申请号: | 201811646113.3 | 申请日: | 2018-12-29 |
| 公开(公告)号: | CN109754084B | 公开(公告)日: | 2020-06-12 |
| 发明(设计)人: | 不公告发明人 | 申请(专利权)人: | 中科寒武纪科技股份有限公司 |
| 主分类号: | G06N3/08 | 分类号: | G06N3/08;G06K9/62 |
| 代理公司: | 北京华进京联知识产权代理有限公司 11606 | 代理人: | 朱五云 |
| 地址: | 100190 北京市海*** | 国省代码: | 北京;11 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 网络 结构 处理 方法 装置 相关 产品 | ||
1.一种网络结构的处理方法,其特征在于,所述包括:
基于Caffe框架下使用检测网络SSD检测图像时,获取优化指令;所述优化指令用于对检测网络SSD的网络结构的结构,或是同时对所述网络结构的结构,以及所述网络结构中卷积层的模型参数进行优化处理;所述检测网络SSD用于对图像进行检测;
根据所述优化指令中的优化级别删除所述网络结构中的冗余层,并在所述网络结构中添加ssd检测层,以对网络结构进行优化处理;其中,所述冗余层是根据实际应用需求或用户对所述网络结构的功能需求确定的;所述冗余层为不需要强制执行的数据变化操作层;所述冗余层包括默认框设置层、拼接层、Permute层、Flatten层中的至少一层;所述ssd检测层用于设置默认框、拼接特征图以及输出检测结果;所述优化级别包括第一优化级别和第二优化级别,所述第一优化级别用于指示计算机设备对所述网络结构进行结构优化的操作,所述第二优化级别用于指示所述计算机设备对所述网络结构进行结构优化的操作,以及对所述网络结构中的卷积层的模型参数进行优化的操作。
2.如权利要求1所述的方法,其特征在于,根据所述优化指令中的优化级别对网络结构进行优化处理的步骤包括:
所述优化级别为第一优化级别时,删除所述网络结构中的冗余层,获得第一优化网络结构。
3.如权利要求1所述的方法,其特征在于,根据所述优化指令中的优化级别对网络结构进行优化处理的步骤包括:
所述优化为第二优化级别时,所述网络结构中存在卷积层和Batch Norm层两层连续的结构、卷积层和Scale层两层连续的结构以及卷积层、Batch Norm层和Scale三层连续的结构中的至少任一种网络结构时,删除所述网络结构中的冗余层,对所述卷积层的模型参数进行优化,获得第二优化网络结构。
4.如权利要求1所述的方法,其特征在于,根据所述优化指令中的优化级别对网络结构进行优化处理的步骤包括:
所述优化级别为第二优化级别时,所述网络结构中不存在卷积层和Batch Norm层两层连续的结构、卷积层和Scale层两层连续的结构以及卷积层、Batch Norm层和Scale三层连续的结构中的任一种网络结构时,删除所述网络结构中的冗余层,获得第一优化网络结构。
5.根据权利要求2、3或4所述的方法,其特征在于,删除所述网络结构中的冗余层,包括:
判断所述网络结构中是否存在检测输出层;
若存在所述检测输出层,则识别所述网络结构中的所述冗余层;
根据识别结果删除所述网络结构中的所述冗余层,并在所述网络结构中添加ssd检测层;其中,所述ssd检测层用于设置默认框、拼接特征图、以及输出检测结果。
6.根据权利要求5所述的方法,其特征在于,判断所述网络结构中是否存在检测输出层,包括:
获取所述网络结构中的最后一层结构层的类型信息;
根据所述类型信息,判断所述最后一层结构层是否为所述检测输出层。
7.根据权利要求5所述的方法,其特征在于,识别所述网络结构中的所述冗余层,包括:
按照预设的查找路径,以所述检测输出层为起点,获取与所述检测输出层连接的所有路径上包含的各层结构层的类型信息;
根据所述各层结构层的类型信息,识别所述网络结构中的冗余层。
8.根据权利要求5所述的方法,其特征在于,删除所述网络结构中的冗余层,还包括:
删除所述网络结构中的冗余层,将所述网络结构中的除所述冗余层以外的其它结构层的连接关系进行更新,得到删除后的网络结构;
将所述删除后的网络结构中的所述检测输出层替换为所述ssd检测层。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于中科寒武纪科技股份有限公司,未经中科寒武纪科技股份有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201811646113.3/1.html,转载请声明来源钻瓜专利网。





