网络模型压缩方法、装置
摘要:
本申请的实施例提供了一种网络模型压缩方法、装置。该方法包括:获取待压缩的网络模型,所述网络模型中包括至少两个滤波器,所述滤波器用于对输入至所述网络模型的图像数据进行特征提取;基于各个滤波器中的滤波器参数,计算任意两个滤波器之间的向量距离;通过任意两个滤波器之间的向量距离,确定所述网络模型中的中心滤波器和非中心滤波器,所述中心滤波器提取的图像特征中包括所述非中心滤波器提取的部分图像特征;对所述非中心滤波器进行剪枝,得到由所述中心滤波器构成的压缩网络模型。本申请实施例的技术方案可以在对网络模型进行压缩的情况下保持模型的精度。
公开/授权文献
0/0