网络混合压缩方法、装置、计算机设备和存储介质
摘要:
本申请涉及一种网络混合压缩方法、装置、计算机设备、存储介质和计算机程序产品。所述方法包括:通过样本图像对目标检测网络进行训练,得到教师网络;对教师网络进行蒸馏处理,得到教师辅助网络;根据教师网络中各滤波器之间的冗余度,对教师网络进行滤波器剪枝,得到剪枝后的教师网络;根据教师辅助网络中各滤波器之间的冗余度,对教师辅助网络进行滤波器剪枝,得到剪枝后的教师辅助网络;根据剪枝后的教师网络的中间层特征和剪枝后的教师辅助网络的中间层特征,对学生网络进行蒸馏,得到蒸馏后的学生网络;蒸馏后的学生网络用于对目标图像进行目标检测处理。采用本方法能够提高了蒸馏得到的学生网络的性能。
0/0