基于改进YOLOv4的垃圾识别研究

摘要

本文首先创建了自制的生活垃圾数据集,并针对垃圾数据集类间相似、类内差异大的问题,提出了一种加入三重注意力机制的改进的YOLOv4算法,提升有效的特征通道权重,加强特征提取能力。将改进后的网络模型在自制的数据集上进行训练、验证和测试,实验结果表明,改进算法可以有效提高垃圾识别的准确率。