结合渐进式引导蒸馏和结构重构的检测网络轻量化方法

专利主权项内容

1.一种结合渐进式引导蒸馏和结构重构的检测网络轻量化方法，其特征在于所述方法包括如下步骤：步骤1：将MobileNet v3 Block中的SE注意力机制替换为CBAM注意力机制，并采用改进后的MobileNet v3 Block结构重构原始检测网络中的卷积模块得到学生网络；步骤2：基于批归一化的剪枝方法精简学生网络，引入批归一化层的缩放系数γ代表卷积层输出的每个通道的重要性，根据缩放系数γ的大小来确定要剪枝的通道；步骤3：选用原始检测网络作为教师网络，对参数相对较多、结构相对复杂的教师网络进行训练；步骤4：调整原始检测网络中的模块堆叠数目和每个网络单元的输出通道数得到不同参数量的助教网络，引入教师网络提供先验知识，使用教师网络主干特征的映射作为提示帮助训练助教网络，再递归地使用参数量大的助教网络蒸馏参数量小的助教网络，直至蒸馏出最后一层助教网络的规模满足轻量化的需求；步骤5：使用步骤4中最后一层助教网络的主干特征映射作为提示，对步骤1、步骤2精简后的学生网络进行蒸馏。