发明公开
- 专利标题: 一种SSD目标检测卷积神经网络的加速计算方法
- 专利标题(英): Accelerated calculation method of SSD target detection convolutional neural network
-
申请号: CN201910846935.4申请日: 2019-09-09
-
公开(公告)号: CN110647988A公开(公告)日: 2020-01-03
- 发明人: 韩梦江 , 楼燚航 , 白燕 , 张永祥 , 陈杰
- 申请人: 博云视觉(北京)科技有限公司
- 申请人地址: 北京市海淀区西北旺东路10号院东区2号楼3层303A室
- 专利权人: 博云视觉(北京)科技有限公司
- 当前专利权人: 博云视觉(北京)科技有限公司
- 当前专利权人地址: 北京市海淀区西北旺东路10号院东区2号楼3层303A室
- 代理机构: 北京众泽信达知识产权代理事务所
- 代理商 张艳萍
- 主分类号: G06N3/08
- IPC分类号: G06N3/08 ; G06N3/04
摘要:
本发明公开了一种SSD目标检测卷积神经网络的加速计算方法,包括以下步骤:S1、将SSD目标检测卷积神经网络训练阶段的目标框Box回归层和目标分类回归层删除,使用一个新的卷积层同时进行Box推导计算和目标分类计算;S2、在新的卷积层后添加一个BlobSplit层,通过BlobSplit层将新的卷积层输出的Blob按照配置的channel拆分成两个独立的Blob并输出给训练阶段后续的两个相关层;S3、将SSD目标检测卷积神经网络推导阶段的目标框Box回归层和目标分类回归层删除,使用一个新的卷积层同时进行Box推导计算和目标分类计算;S4、将SSD目标检测卷积神经网络推导阶段的后续相关层全部删除,在新的卷积层后连接一个Object_Detector层。