一种卷积神经网络的按通道L2范数聚类剪枝方法

摘要

本发明公开了一种卷积神经网络的按通道L2范数聚类剪枝方法及系统，属于剪枝技术领域，包括以下步骤：S1：获取预训练模型；S2：通道K‑Means二聚类；S3：通道裁剪；S4：裁剪结束判断。本发明通过评价卷积核中每个通道的重要性，然后去除不重要的通道，随后进行剪枝模型训练，而后对剪枝是否结束进行判断，完成剪枝后对模型微调并保存，简化了神经网络的计算数据，对深层卷积神经网络的权重起到压缩的作用，大大缩减了存储空间及计算量，值得被推广使用。