首页|期刊导航|计算机工程与应用|基于YOLOv3的卷积层结构化剪枝

基于YOLOv3的卷积层结构化剪枝OA北大核心CSCDCSTPCD

Convolutional Layered Pruning Based on YOLOv3

中文摘要

针对YOLOv3等卷积神经网络使用更多的卷积层结构且卷积核尺寸大小统一的特点,提出一种卷积层结构化剪枝计算的参数压缩方法.基于卷积层权值参数,设计度量卷积层重要性公式,评估卷积层相对整个网络重要性,计算卷积层重要度并对得分进行排序,制定稀疏值分配策略,再训练模型操作保证模型性能不降低,并得到各个卷积层分配的稀疏值以及卷积过滤器,完成模型的结构化剪枝计算.在Darknet上实现YOLOv3卷积层结构化剪枝的参数压缩方法,不仅将YOLOv3参数压缩1…查看全部>>

张良;张增;舒伟华;梅魁志

武汉数字工程研究所,武汉 430074西安交通大学 电子与信息工程学部,西安 710049西安交通大学 电子与信息工程学部,西安 710049中国航发南方工业有限公司,湖南 株洲 412002

信息技术与安全科学

深度学习剪枝卷积神经网络

《计算机工程与应用》 2021 (6)

131-137,7

10.3778/j.issn.1002-8331.1912-0302

评论

您当前未登录!去登录点击加载更多...