一种SSD目标检测卷积神经网络的加速计算方法
摘要:
本发明公开了一种SSD目标检测卷积神经网络的加速计算方法,包括以下步骤:S1、将SSD目标检测卷积神经网络训练阶段的目标框Box回归层和目标分类回归层删除,使用一个新的卷积层同时进行Box推导计算和目标分类计算;S2、在新的卷积层后添加一个BlobSplit层,通过BlobSplit层将新的卷积层输出的Blob按照配置的channel拆分成两个独立的Blob并输出给训练阶段后续的两个相关层;S3、将SSD目标检测卷积神经网络推导阶段的目标框Box回归层和目标分类回归层删除,使用一个新的卷积层同时进行Box推导计算和目标分类计算;S4、将SSD目标检测卷积神经网络推导阶段的后续相关层全部删除,在新的卷积层后连接一个Object_Detector层。
0/0