融合GhostNet的YOLOv5垃圾分類方法
所屬分類:技術論文
上傳者:wwei
文檔大?。?span>5286 K
標簽: 垃圾分類 ECA GhostNet
所需積分:0分積分不夠怎么辦?
文檔介紹:垃圾分類是建設生態(tài)文明的重要一環(huán),為解決重量級模型難以部署移動端設備的問題,提出基于YOLOv5網(wǎng)絡改進的垃圾圖像分類方法。采用融合GhostNet的主干網(wǎng)絡,用線性運算代替?zhèn)鹘y(tǒng)卷積運算,降低了模型的參數(shù)量,提高了模型推理速度;通過在網(wǎng)絡中加入改進版通道注意力模塊,強化重要的通道特征,獲取更多深層次的特征信息;采用加權邊界融合方法,提升檢測框的定位精度。經實驗證明,該方法在自制數(shù)據(jù)集中較原模型的精度提高了8.5%,參數(shù)量減少了46.7%,平均推理速度提高了1.22 ms,實現(xiàn)了精度和推理速度的綜合提升。 垃圾分類;ECA;GhostNet;YOLOv5
現(xiàn)在下載
VIP會員,AET專家下載不扣分;重復下載不扣分,本人上傳資源不扣分。