典型文献
面向垃圾图像分类的改进注意力机制剪枝算法
文献摘要：
为实现垃圾图像分类模型在嵌入式设备上的部署,提高垃圾分类处理效率,文章提出一种基于改进注意力机制的剪枝算法.首先在空间注意力模块,对特征图沿行、列两个方向进行评价,然后在通道注意力模块,沿特征图通道方向进行评价,最后形成综合注意力评价向量.根据改进的注意力机制形成的通道注意力向量对通道重要性进行评价,指导模型进行剪枝.在公开数据集TrashNet和文中构建的生活垃圾数据集上对ResNet50和VGG16进行基于改进注意力机制的剪枝实验,结果表明,本文提出的剪枝算法可以在较小的准确率损失下对模型进行有效压缩.
文献关键词：
垃圾分类;注意力机制;通道剪枝
中图分类号：
作者姓名：
高文晓
作者机构：
北方民族大学电气信息工程学院 宁夏 银川 750021
文献出处：
引用格式：
[1]高文晓-.面向垃圾图像分类的改进注意力机制剪枝算法)[J].科学与信息化,2022(24):60-63
A类：
TrashNet
B类：
图像分类,改进注意力机制,剪枝算法,分类模型,嵌入式设备,垃圾分类,分类处理,处理效率,空间注意力,特征图,沿行,通道注意力模块,公开数据集,生活垃圾,ResNet50,VGG16,通道剪枝
AB值：
0.292622
相似文献
机标中图分类号，由域田数据科技根据网络公开资料自动分析生成，仅供学习研究参考。
