基于GAM-YOLOv8算法的生活垃圾检测

Domestic Waste Detection Based on GAM-YOLOv8 Algorithm

摘要: 日常生活垃圾分拣是困扰人们的一个难题，生活垃圾的种类繁多，所处环境复杂，常用的目标检测算法无法适应各种复杂的环境，导致精度较低。为了准确的分拣生活垃圾，提出了一种基于GAM注意力机制的YOLOv8生活垃圾检测算法。该算法在YOLOv8优秀的目标检测基础上，加入GAM注意力机制，增强网络对重要通道特征信息的关注能力，提升高层网络中图像特征语义信息的提取能力，提高复杂环境垃圾分类检测精度的效果。实验表明，在40多种生活垃圾类别检测测试中，改进的YOLOv8算法mAP平均精度84.5%，较原始算法YOLOv8提升了0.7%。因此改进的YOLOv8算法可以通过对垃圾图像的分析和识别，帮助人们准确地进行垃圾分类。较好的满足了生活垃圾检测精度的要求。

Abstract: The sorting of daily household garbage is a difficult problem that troubles people. There are many kinds of household garbage and the environment is complicated. In order to sort domestic waste accurately, a YOLOv8 domestic waste detection algorithm based on GAM attention mechanism was proposed. Based on the excellent target detection of YOLOv8, this algorithm adds GAM attention mechanism to enhance the network’s ability to pay attention to important channel feature information and improve the extraction ability of image feature semantic information in high-level networks. Experiments show that in more than 40 types of domestic waste detection tests, the average accuracy of the improved YOLOv8 algorithm mAP is 84.5%, which is 0.7% higher than that of the original YOLOv8 algorithm. Therefore, the improved YOLOv8 algorithm can help people to classify garbage quickly and accurately through the analysis and recognition of garbage images. It can meet the requirement of detecting precision of domestic waste.