科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明实施例公开了一种检测模型压缩方法、装置和计算机可读存储介质,对获取的待检测图片按照预设的图片尺寸进行缩放处理,得到标准图像；将各标准图像输入初始检测模型,计算初始检测模型中特定卷积层中各个通道的损失梯度值。损失梯度值越小,说明通道的表征能力越弱。根据各个通道的损失梯度值以及特定卷积层对应的删除比例,对各所述特定卷积层进行剪枝处理,得到压缩后的检测模型。通过对检测模型进行压缩,大大减少了检测模型对内存的要求,可以使得压缩后的检测模型很方便的移植到存储和计算资源有限的硬件设备中。通过删除检测模型中表征能力较弱的通道,减少了检测模型的网络参数,提升了检测模型的计算效率。
- 单位
