融合GhostNet的YOLOv5垃圾分类方法
电子技术应用
李耀，胡军国，乐杨
浙江农林大学 数学与计算机科学学院，浙江 杭州 311300
摘要： 垃圾分类是建设生态文明的重要一环，为解决重量级模型难以部署移动端设备的问题，提出基于YOLOv5网络改进的垃圾图像分类方法。采用融合GhostNet的主干网络，用线性运算代替传统卷积运算，降低了模型的参数量，提高了模型推理速度；通过在网络中加入改进版通道注意力模块，强化重要的通道特征，获取更多深层次的特征信息；采用加权边界融合方法，提升检测框的定位精度。经实验证明，该方法在自制数据集中较原模型的精度提高了8.5%，参数量减少了46.7%，平均推理速度提高了1.22 ms，实现了精度和推理速度的综合提升。
中圖分類號：TP391 文獻標志碼：A DOI: 10.16157/j.issn.0258-7998.234428
中文引用格式： 李耀，胡軍國，樂楊. 融合GhostNet的YOLOv5垃圾分類方法[J]. 電子技術應用，2024，50(1)：14-20.
英文引用格式： Li Yao，Hu Junguo，Le Yang. YOLOv5 garbage classification method with GhostNet[J]. Application of Electronic Technique，2024，50(1)：14-20.
中文引用格式： 李耀，胡軍國，樂楊. 融合GhostNet的YOLOv5垃圾分類方法[J]. 電子技術應用，2024，50(1)：14-20.
英文引用格式： Li Yao，Hu Junguo，Le Yang. YOLOv5 garbage classification method with GhostNet[J]. Application of Electronic Technique，2024，50(1)：14-20.
YOLOv5 garbage classification method with GhostNet
Li Yao，Hu Junguo，Le Yang
School of Mathematics and Computer Science， Zhejiang A&F University， Hangzhou 311300， China
Abstract： Garbage classification is an important part of building ecological civilization. To solve the problem that heavyweight models are difficult to deploy to mobile devices, an improved garbage image classification method based on YOLOv5 network is proposed. The backbone network fused with GhostNet is used to replace the traditional convolutional operation with linear operation, which reduces the number of parameters of the model and improves the model inference speed. By adding an improved version of channel attention module to the network, the important channel features are strengthened and more deep-level feature information is obtained. The weighted boundary fusion method is used to improve the localization accuracy of the detection frame. It is experimentally demonstrated that the method improves the accuracy by 8.5%, reduces the parameter quantity by 46.7%, and increases the average inference speed by 1.22 ms in the homemade dataset compared with the original model, achieving a comprehensive improvement of accuracy and inference speed.
Key words : garbage classification；ECA；GhostNet；YOLOv5
引言
近年來，隨著我國城市人口的增加，城市生活垃圾總量增長迅速。據相關統計，我國城市生活垃圾清運量已經從2004年的15 509萬噸增長為2021年的24 869萬噸。自黨的十八大以來，黨中央高度重視生態文明建設，垃圾處理問題已經成為城市生活中必須解決的問題，在城市社區生活中，居民垃圾分類的意識較弱，且較多依賴于傳統人工分揀，傳統人工分揀存在耗時長、效率低、工作環境差及精準度較低等問題。若有效地利用計算機視覺技術對垃圾進行目標檢測，通過對垃圾目標的快速定位和精確分類，將極大地減少人力資源的消耗，有效地提高垃圾分類效率，為后續的垃圾回收工作提供支持，進一步改善城市人居環境。
本文詳細內容請下載：
http://www.tom3567.com/resource/share/2000005827
作者信息：
李耀，胡軍國，樂楊
（浙江農林大學 數學與計算機科學學院，浙江 杭州 311300）
此內容為AET網站原創，未經授權禁止轉載。